imp no structure

smith-nathanh · Oct 25, 2024 · 020bee0 · 020bee0
1 parent 6bc5984
commit 020bee0
Show file tree

Hide file tree

Showing 2 changed files with 312 additions and 2 deletions.
diff --git a/graph.py b/graph.py
@@ -118,9 +118,9 @@ def implementation(state: GraphState):
                                           architecture_design=state["documents"]["architecture_design"],
                                           requirements=state["documents"]["requirements"])
     structured_llm = llm.with_structured_output(Implementation)
-    code = structured_llm.invoke([HumanMessage(content=prompt)]) # use structured outputs here
+    code = llm.invoke([HumanMessage(content=prompt)]) # use structured outputs here
     #code = {"file1": "code", "file2": "code"} # temporary response
-    state['documents'].update(code.dict())
+    state['documents'].update({"code": code.content})
     return state
 
 def approve_implementation(state: GraphState):