knowledge graph

knowledge.py

@@ -1,4 +1,4 @@
-from openai import OpenAI
+
 import instructor
 
 from pydantic import BaseModel, Field
@@ -37,6 +37,7 @@ client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 # Enable instructor patches for Groq client
 client = instructor.from_groq(client)
 """
+from openai import OpenAI
 client = instructor.from_openai(
     OpenAI(
         base_url="http://localhost:11434/v1",
@@ -76,4 +77,4 @@ def visualize_knowledge_graph(kg: KnowledgeGraph):
 def graph(query):
     graph = generate_graph(query)
     visualize_knowledge_graph(graph)
-    return "./knowledge_graph.png"
+    return "knowledge_graph.png"