Skip to content

Commit

Permalink
fix: Use bind_tools API to bind all tools with LLM.
Browse files Browse the repository at this point in the history
This is required to use tool calling feature.
  • Loading branch information
anubhav756 committed Nov 15, 2024
1 parent c36c7fb commit 80e73dd
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,7 @@ def initialize_agent(
model: str,
) -> "UserAgent":
llm = ChatVertexAI(max_output_tokens=512, model_name=model, temperature=0.0)
llm.bind_tools(tools)
memory = ConversationBufferMemory(
chat_memory=ChatMessageHistory(messages=history),
memory_key="chat_history",
Expand Down

0 comments on commit 80e73dd

Please sign in to comment.