bstraehle commited on
Commit
c0f5f7a
1 Parent(s): 29adcdb

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. rag_langgraph.py +4 -4
rag_langgraph.py CHANGED
@@ -99,7 +99,7 @@ def create_graph(topic):
99
  researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt="1. Research content on topic: " + topic + ", prioritizing research papers. "
100
  "2. Based on your research, write a 2000-word article on the topic. "
101
  "3. At the beginning of the article, add current date and author: Multi-AI-Agent System. "
102
- "4. At the end of the article, add a reference section with research papers.")
103
  researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
104
 
105
  workflow = StateGraph(AgentState)
@@ -125,7 +125,7 @@ def run_multi_agent(topic):
125
  ]
126
  })
127
  article = result['messages'][-1].content
128
- print("###")
129
- print(article)
130
- print("###")
131
  return article
 
99
  researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt="1. Research content on topic: " + topic + ", prioritizing research papers. "
100
  "2. Based on your research, write a 2000-word article on the topic. "
101
  "3. At the beginning of the article, add current date and author: Multi-AI-Agent System. "
102
+ "4. At the end of the article, add a references section with research papers.")
103
  researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
104
 
105
  workflow = StateGraph(AgentState)
 
125
  ]
126
  })
127
  article = result['messages'][-1].content
128
+ #print("***")
129
+ #print(article)
130
+ #print("***")
131
  return article