Update app.py

app.py

@@ -13,11 +13,10 @@ def generate_knowledge_graph(api_key, user_input):
     completion = openai.ChatCompletion.create(
         model="gpt-3.5-turbo-16k",
         messages=[
-            {"role": "system", "content": "You are a helpful assistant."},
             {
                 "role": "user",
                 "content": f"Help me understand following by describing as a detailed knowledge graph: {user_input}",
-            },
+            }
         ],
         functions=[
             {
@@ -83,9 +82,11 @@ def generate_knowledge_graph(api_key, user_input):
                     "required": ["nodes", "edges"],
                 },
             }
-        ]
+        ],
+        function_call={"name": "knowledge_graph"},
     )
-    response_data = completion.choices[0]["message"]["function_results"]["knowledge_graph"]
+
+    response_data = completion.choices[0]["message"]["function_call"]["arguments"]
     print(response_data)
 
     # Visualizar o conhecimento usando Graphviz