From b9570b2eb9c9926248fb7a882d83b7bc314f68ae Mon Sep 17 00:00:00 2001
From: Marcus Schiesser <mail@marcusschiesser.de>
Date: Mon, 4 Nov 2024 11:34:13 +0700
Subject: [PATCH] fix: use generic LLMAgent instead of OpenAIAgent (adds
support for Gemini and Anthropic for Agentic RAG) (#410)
---
.changeset/shy-bulldogs-wait.md | 5 +++++
templates/components/engines/typescript/agent/chat.ts | 4 ++--
2 files changed, 7 insertions(+), 2 deletions(-)
create mode 100644 .changeset/shy-bulldogs-wait.md
diff --git a/.changeset/shy-bulldogs-wait.md b/.changeset/shy-bulldogs-wait.md
new file mode 100644
index 00000000..1da10248
--- /dev/null
+++ b/.changeset/shy-bulldogs-wait.md
@@ -0,0 +1,5 @@
+---
+"create-llama": patch
+---
+
+Fix: use generic LLMAgent instead of OpenAIAgent (adds support for Gemini and Anthropic for Agentic RAG)
diff --git a/templates/components/engines/typescript/agent/chat.ts b/templates/components/engines/typescript/agent/chat.ts
index bf2bb69b..333ff640 100644
--- a/templates/components/engines/typescript/agent/chat.ts
+++ b/templates/components/engines/typescript/agent/chat.ts
@@ -1,7 +1,7 @@
import {
BaseChatEngine,
BaseToolWithCall,
- OpenAIAgent,
+ LLMAgent,
QueryEngineTool,
} from "llamaindex";
import fs from "node:fs/promises";
@@ -42,7 +42,7 @@ export async function createChatEngine(documentIds?: string[], params?: any) {
tools.push(...(await createTools(toolConfig)));
}
- const agent = new OpenAIAgent({
+ const agent = new LLMAgent({
tools,
systemPrompt: process.env.SYSTEM_PROMPT,
}) as unknown as BaseChatEngine;
--
GitLab