Fix bugs

trustgraph/chunker/recursive/chunker.py

@@ -4,6 +4,9 @@ Simple decoder, accepts text documents on input, outputs chunks from the
 as text as separate output objects.
 """
 
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+
+
 from ... schema import TextDocument, Chunk, Source
 from ... log_level import LogLevel
 from ... base import ConsumerProducer

trustgraph/decoder/pdf/pdf_decoder.py

@@ -4,6 +4,8 @@ Simple decoder, accepts PDF documents on input, outputs pages from the
 PDF document as text as separate output objects.
 """
 
+import tempfile
+import base64
 from langchain_community.document_loaders import PyPDFLoader
 
 from ... schema import Document, TextDocument, Source

trustgraph/kg/extract_definitions/extract.py

@@ -4,6 +4,9 @@ Simple decoder, accepts vector+text chunks input, applies entity analysis to
 get entity definitions which are output as graph edges.
 """
 
+import urllib.parse
+import json
+
 from ... schema import VectorsChunk, Triple, Source, Value
 from ... log_level import LogLevel
 from ... llm_client import LlmClient

trustgraph/kg/extract_relationships/extract.py

@@ -5,6 +5,8 @@ relationship analysis to get entity relationship edges which are output as
 graph edges.
 """
 
+import urllib.parse
+import json
 from pulsar.schema import JsonSchema
 
 from ... schema import VectorsChunk, Triple, VectorsAssociation, Source, Value
@@ -84,8 +86,6 @@ class Processor(ConsumerProducer):
 
         chunk = v.chunk.decode("utf-8")
 
-        g = rdflib.Graph()
-
         try:
 
             rels = self.get_relationships(chunk)

trustgraph/rag/graph/rag.py

@@ -37,8 +37,8 @@ class Processor(ConsumerProducer):
             input_queue=input_queue,
             output_queue=output_queue,
             subscriber=subscriber,
-            input_schema=TextCompletionRequest,
-            output_schema=TextCompletionResponse,
+            input_schema=GraphRagQuery,
+            output_schema=GraphRagResponse,
         )
 
         self.rag = GraphRag(