1212import com .embabel .agent .core .CoreToolGroups ;
1313import com .embabel .agent .discord .DiscordUser ;
1414import com .embabel .agent .rag .neo .drivine .DrivineStore ;
15- import com .embabel .agent .rag .neo . drivine . GranularRagReference ;
16- import com .embabel .agent .rag .tools .RagReference ;
15+ import com .embabel .agent .rag .tools . RagServiceReference ;
16+ import com .embabel .agent .rag .tools .ToolishRag ;
1717import com .embabel .chat .AssistantMessage ;
1818import com .embabel .chat .Chatbot ;
1919import com .embabel .chat .Conversation ;
@@ -112,12 +112,12 @@ ConversationStatus respond(
112112 var templateModel = new HashMap <String , Object >();
113113
114114 templateModel .put ("persona" , persona );
115- var ragReference = new RagReference ("docs" ,
115+ var ragReference = new RagServiceReference ("docs" ,
116116 "Embabel docs" ,
117117 guideProperties .ragOptions (dataManager .embabelContentRagServiceFor (context )),
118118 context .ai ().withLlmByRole ("summarizer" ));
119119
120- var granularRagReference = new GranularRagReference (
120+ var toolishRag = new ToolishRag (
121121 "granular_docs" ,
122122 "Embabel docs with granular retrieval" ,
123123 drivineStore
@@ -130,7 +130,7 @@ ConversationStatus respond(
130130 .withReferences (dataManager .referencesForUser (context .user ()))
131131 .withTools (CoreToolGroups .WEB )
132132// .withReference(ragReference)
133- .withReferences (granularRagReference )
133+ .withReferences (toolishRag )
134134 .withTemplate ("guide_system" )
135135 .respondWithSystemPrompt (conversation , templateModel );
136136 conversation .addMessage (assistantMessage );
0 commit comments