adrienbrdne commited on
Commit
4f7caa6
·
verified ·
1 Parent(s): 3a293cc

Update ki_gen/planner.py

Browse files
Files changed (1) hide show
  1. ki_gen/planner.py +4 -5
ki_gen/planner.py CHANGED
@@ -64,11 +64,10 @@ def chatbot_llama(state: State):
64
  except Exception as error:
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):
 
64
  except Exception as error:
65
  error_chatbot_groq(error,llm_llama,state["messages"])
66
  def chatbot_mixtral(state: State):
67
+ print(state)
68
+ llm_mixtral = ChatGroq(model="deepseek-r1-distill-llama-70b")
69
+ print(llm_mixtral)
70
+ return {"messages" : [llm_mixtral.invoke(state["messages"])]}
 
71
  # except Exception as error:
72
  # error_chatbot_groq(error,llm_mixtral,state["messages"])
73
  def chatbot_openai(state: State):