Fixed bug with edge handler

kg_summarizer/server.py

@@ -9,7 +9,7 @@
 
 
 class LLMParameters(BaseModel):
-    gpt_model: str
+    gpt_model: str = "gpt-3.5-turbo"
     temperature: Optional[float] = 0.0
     system_prompt: Optional[str] = ""
 
@@ -38,7 +38,7 @@ class EdgeItem(BaseModel):
     parameters: Parameters
 
 
-KG_SUM_VERSION = "0.0.4"
+KG_SUM_VERSION = "0.0.5"
 
 # declare the application and populate some details
 app = FastAPI(
@@ -64,7 +64,7 @@ async def summarize_abstract_handler(item: AbstractItem):
 
 @app.post("/summarize/edge")
 async def summarize_edge_handler(item: EdgeItem):
-    edge = EdgeContainer(item)
+    edge = EdgeContainer(item.edge)
 
     spo_sentence = edge.format_spo_sentence()
 

setup.py

@@ -8,7 +8,7 @@
 
 setup(
     name="kg-summarizer",
-    version="0.0.4",
+    version="0.0.5",
     author="Joey Richardson",
     author_email="[email protected]",
     url="https://github.com/jrichardson97/kg-summarizer",