openai.BadRequestError: Error code: 400 - {'object': 'error', 'message': 'Tool call IDs should be alphanumeric strings with length 9!', 'type': 'BadRequestError', 'param': None, 'code': 400} #29021
Labels
🤖:bug
Related to a bug, vulnerability, unexpected error with an existing feature
Comments
dosubot
bot
added
the
🤖:bug
|
@RachelShalom I have optimized your code and all is working well, please check: import os
import random
import string
from math import factorial
from langgraph.graph import StateGraph, END
from typing import TypedDict, Annotated
import operator
from langchain_core.messages import AnyMessage,SystemMessage, HumanMessage , ToolMessage
from pydantic import BaseModel
from langchain_mistralai import ChatMistralAI
from langchain_community.tools.tavily_search import TavilySearchResults
from langchain_community.tools.jira.tool import JiraAction
from langchain.agents.tool_calling_agent.base import create_tool_calling_agent
import httpx
http_client=httpx.Client(verify=False)
#create the tool
tool = TavilySearchResults(max_results=2)
print(type(tool))
print(tool.name)
#create an agent state: an annotated list of messages
class AgentState(TypedDict):
messages: Annotated[list[AnyMessage], operator.add]
class Agent:
def __init__(self, model, tools, system =""):
self.system = system
graph = StateGraph(AgentState)
graph.add_node("llm", self.call_model)
graph.add_node("action", self.take_action)
graph.add_conditional_edges(
"llm",
self.exists_action,
{True:"action", False:END}
)
graph.add_edge("action", "llm")
graph.set_entry_point("llm")
self.graph = graph.compile() # cpm[ile turns the graph into a langchain runnable
self.tools = {t.name: t for t in tools}
self.model = model.bind_tools(tools, tool_choice="auto") # Change this to auto
def exists_action(self, state:AgentState):
result = state['messages'][-1]
return len(result.tool_calls) > 0
def call_model(self, state:AgentState):
messages = state["messages"]
if self.system:
messages = [SystemMessage(content=self.system)] + messages
message = self.model.invoke(messages)
return {'messages':[message]}
def take_action(self, state:AgentState):
tool_calls = state['messages'][-1].tool_calls
results = []
for t in tool_calls:
result = self.tools[t['name']].invoke(t['args'])
results.append(ToolMessage(tool_call_id=t['id'],name = t['name'] ,content=str(result)))
return {'messages': results}
@staticmethod
def generate_tool_call_id(length=9):
return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
prompt = """
you are a smart research assistant use the search engine to look up information. \
you are allowed to make multiple calls(either together or in a sequence).
only look for information when you are sure of what you want .\
If you need to look up some information before asking a follow up questions, you are allowed to do that!
"""
base_url=os.getenv("BASE_URL")
key = os.getenv("DEVGENAI_KEY")
mistral_llm = ChatMistralAI(model ="mistral-large-latest", top_p = 1, temperature = 0.5, verbose = True,) # Use MistralAI Client directly.
SearchAgent = Agent(mistral_llm, [tool], system = f"[INST]{prompt}[/INST]")
messages = [HumanMessage(content = "what's the weather in Boston, MA?")]
result = SearchAgent.graph.invoke({"messages":messages})
print(result)
print(result["messages"][-1].content) |
|
Hey @keenborder786 thank for this I see you changed the model for the mistralAI class. I am using a local endpoint ( not mistral AI) which does not seem to work well with mistralAIchatbot class |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Checked other resources
Example Code
The text was updated successfully, but these errors were encountered: