Skip to content

Commit 81ea1d4

Browse files
committed
Update to agentic RAG
1 parent 74df5f2 commit 81ea1d4

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
<dependency>
3535
<groupId>com.embabel.agent</groupId>
3636
<artifactId>embabel-agent-rag-neo-drivine</artifactId>
37-
<version>0.1.0-SNAPSHOT</version>
37+
<version>0.1.1-SNAPSHOT</version>
3838
</dependency>
3939

4040
<dependency>

src/main/java/com/embabel/guide/GuideResponderAgent.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@
1212
import com.embabel.agent.core.CoreToolGroups;
1313
import com.embabel.agent.discord.DiscordUser;
1414
import 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;
1717
import com.embabel.chat.AssistantMessage;
1818
import com.embabel.chat.Chatbot;
1919
import 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

Comments
 (0)