knowledge graph
Browse files- app.py +2 -10
- knowledge.py +3 -2
app.py
CHANGED
|
@@ -4,16 +4,8 @@ from tool import rival_product
|
|
| 4 |
from graphrag import reasoning
|
| 5 |
from knowledge import graph
|
| 6 |
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
'''
|
| 10 |
-
import pathlib
|
| 11 |
-
import platform
|
| 12 |
-
plt = platform.system()
|
| 13 |
-
if plt == 'Windows':
|
| 14 |
-
pathlib.PosixPath = pathlib.WindowsPath
|
| 15 |
-
else:
|
| 16 |
-
pathlib.WindowsPath = pathlib.PosixPath
|
| 17 |
|
| 18 |
with gr.Blocks() as demo:
|
| 19 |
with gr.Tab("RAG"):
|
|
|
|
| 4 |
from graphrag import reasoning
|
| 5 |
from knowledge import graph
|
| 6 |
|
| 7 |
+
|
| 8 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
with gr.Blocks() as demo:
|
| 11 |
with gr.Tab("RAG"):
|
knowledge.py
CHANGED
|
@@ -31,6 +31,7 @@ class KnowledgeGraph(BaseModel):
|
|
| 31 |
|
| 32 |
from groq import Groq
|
| 33 |
import os
|
|
|
|
| 34 |
# Initialize with API key
|
| 35 |
client = Groq(api_key=os.getenv("GROQ_API_KEY"))
|
| 36 |
|
|
@@ -72,9 +73,9 @@ def visualize_knowledge_graph(kg: KnowledgeGraph):
|
|
| 72 |
dot.edge(str(edge.source), str(edge.target), label=edge.label, color=edge.color)
|
| 73 |
|
| 74 |
# Render the graph
|
| 75 |
-
dot.render("knowledge_graph", format="png")
|
| 76 |
|
| 77 |
def graph(query):
|
| 78 |
graph = generate_graph(query)
|
| 79 |
visualize_knowledge_graph(graph)
|
| 80 |
-
return "knowledge_graph.png"
|
|
|
|
| 31 |
|
| 32 |
from groq import Groq
|
| 33 |
import os
|
| 34 |
+
os.environ["PATH"] += os.pathsep + '/usr/bin/'
|
| 35 |
# Initialize with API key
|
| 36 |
client = Groq(api_key=os.getenv("GROQ_API_KEY"))
|
| 37 |
|
|
|
|
| 73 |
dot.edge(str(edge.source), str(edge.target), label=edge.label, color=edge.color)
|
| 74 |
|
| 75 |
# Render the graph
|
| 76 |
+
dot.render("/tmp/knowledge_graph", format="png")
|
| 77 |
|
| 78 |
def graph(query):
|
| 79 |
graph = generate_graph(query)
|
| 80 |
visualize_knowledge_graph(graph)
|
| 81 |
+
return "/tmp/knowledge_graph.png"
|