adityamanwatkar commited on
Commit
916083b
·
verified ·
1 Parent(s): 75a87aa

Upload 2 files

Browse files
Files changed (2) hide show
  1. src/.env +3 -0
  2. src/main.py +9 -9
src/.env ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ OPENAI_API_KEY=94e1534c-a937-48f4-a715-a721248cd-2b5
2
+ MODEL=gpt-4
3
+ MAX_ITERATIONS=2
src/main.py CHANGED
@@ -1,10 +1,10 @@
1
- from app.graph.workflow import build_graph
2
-
3
- graph = build_graph()
4
-
5
- def run_agent(prompt):
6
- result = graph.invoke({
7
- "prompt": prompt,
8
- "iteration": 0
9
- })
10
  return result
 
1
+ from app.graph.workflow import build_graph
2
+
3
+ graph = build_graph()
4
+
5
+ def run_agent(prompt):
6
+ result = graph.invoke({
7
+ "prompt": prompt,
8
+ "iteration": 0
9
+ })
10
  return result