From 67fab0620bf226cda58be1d1013a91e4bfcdc40a Mon Sep 17 00:00:00 2001 From: Joey Richardson Date: Thu, 25 Jan 2024 14:18:20 -0500 Subject: [PATCH] Fixed bug with edge handler --- kg_summarizer/server.py | 6 +++--- setup.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/kg_summarizer/server.py b/kg_summarizer/server.py index 5067bce..fac51c4 100644 --- a/kg_summarizer/server.py +++ b/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() diff --git a/setup.py b/setup.py index 6e5f0fb..1a38619 100644 --- a/setup.py +++ b/setup.py @@ -8,7 +8,7 @@ setup( name="kg-summarizer", - version="0.0.4", + version="0.0.5", author="Joey Richardson", author_email="richardson.joey.b@gmail.com", url="https://github.com/jrichardson97/kg-summarizer",