Refactoring

airtai · Nov 9, 2023 · 1c28aea · 1c28aea
1 parent 1c162fd
commit 1c28aea
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/application.py b/application.py
@@ -10,11 +10,11 @@
 from google.protobuf import json_format
 from prisma import Prisma
 
-from openai_agent import main
+from openai_agent import openai
 
 app = FastAPI()
 app.include_router(
-    main.router,
+    openai.router,
     tags=["openai"]
 )
 

diff --git a/openai_agent/main.py → openai_agent/openai.py b/openai_agent/main.py → openai_agent/openai.py
@@ -15,4 +15,4 @@ class AzureOpenAIRequest(BaseModel):
 async def create_item(request: AzureOpenAIRequest) -> str:
     conversation = request.conversation
     result = await get_openai_response(conversation)
-    return result
+    return result
diff --git a/openai_agent/utils.py b/openai_agent/utils.py
@@ -47,4 +47,4 @@ async def get_openai_response(conversation: List[Dict[str, str]]) -> str:
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Internal server error: {e}")
     result = completion.choices[0].message.content
-    return result
+    return result