adrienbrdne commited on
Commit
3a293cc
·
verified ·
1 Parent(s): 089de3f

Update ki_gen/planner.py

Browse files
Files changed (1) hide show
  1. ki_gen/planner.py +4 -2
ki_gen/planner.py CHANGED
@@ -65,10 +65,12 @@ def chatbot_llama(state: State):
65
  error_chatbot_groq(error,llm_llama,state["messages"])
66
  def chatbot_mixtral(state: State):
67
  try:
 
68
  llm_mixtral = ChatGroq(model="deepseek-r1-distill-llama-70b")
 
69
  return {"messages" : [llm_mixtral.invoke(state["messages"])]}
70
- except Exception as error:
71
- error_chatbot_groq(error,llm_mixtral,state["messages"])
72
  def chatbot_openai(state: State):
73
  llm_openai = ChatOpenAI(model='gpt-4o', base_url="https://llm.synapse.thalescloud.io/")
74
  return {"messages" : [llm_openai.invoke(state["messages"])]}
 
65
  error_chatbot_groq(error,llm_llama,state["messages"])
66
  def chatbot_mixtral(state: State):
67
  try:
68
+ print(state)
69
  llm_mixtral = ChatGroq(model="deepseek-r1-distill-llama-70b")
70
+ print(llm_mixtral)
71
  return {"messages" : [llm_mixtral.invoke(state["messages"])]}
72
+ # except Exception as error:
73
+ # error_chatbot_groq(error,llm_mixtral,state["messages"])
74
  def chatbot_openai(state: State):
75
  llm_openai = ChatOpenAI(model='gpt-4o', base_url="https://llm.synapse.thalescloud.io/")
76
  return {"messages" : [llm_openai.invoke(state["messages"])]}