bstraehle commited on
Commit
d733788
1 Parent(s): a9795d3

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. rag_langgraph.py +2 -2
rag_langgraph.py CHANGED
@@ -91,7 +91,7 @@ def run_multi_agent(prompt):
91
  ]
92
  ).partial(options=str(options), members=", ".join(members))
93
 
94
- llm = ChatOpenAI(model="gpt-4-1106-preview")
95
 
96
  supervisor_chain = (
97
  prompt
@@ -131,7 +131,7 @@ def run_multi_agent(prompt):
131
  for s in graph.stream(
132
  {
133
  "messages": [
134
- HumanMessage(content="Code hello world and print it to the terminal")
135
  ]
136
  }
137
  ):
 
91
  ]
92
  ).partial(options=str(options), members=", ".join(members))
93
 
94
+ llm = ChatOpenAI(model="gpt-4o")
95
 
96
  supervisor_chain = (
97
  prompt
 
131
  for s in graph.stream(
132
  {
133
  "messages": [
134
+ HumanMessage(content=prompt)
135
  ]
136
  }
137
  ):