Skip to content

Commit 04ad3c6

Browse files
committed
gemini base chat function
1 parent 30698c9 commit 04ad3c6

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

config.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
{
2-
"ChatFunctionName": ""
2+
"ChatFunctionName": "testChatFunction"
33
}

src/agents/base_agent/base_agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ class State(TypedDict):
3535

3636
class BaseAgent:
3737
def __init__(self):
38-
llm = OpenAILLMs() # OpenAILLMs() or GoogleAILLMs()
38+
llm = GoogleAILLMs() # OpenAILLMs() or GoogleAILLMs()
3939
self.llm = llm.get_llm()
40-
summarisation_llm = OpenAILLMs() # OpenAILLMs() or GoogleAILLMs()
40+
summarisation_llm = GoogleAILLMs() # OpenAILLMs() or GoogleAILLMs()
4141
self.summarisation_llm = summarisation_llm.get_llm()
4242
self.summary = ""
4343
self.conversationalStyle = ""

0 commit comments

Comments
 (0)