update apis to generate sow from knowledge base

src/controllers/_sow_controller.py

@@ -112,13 +112,13 @@ class SOWController:
             response_model=ResponseSOWSingle,
             tags=["Upload Requirement File"],
         )
-        self.router.add_api_route(
-            "/sows/{sow_id}/upload",
-            self.upload_sow_by_id,
-            methods=["POST"],
-            response_model=ResponseSOWSingle,
-            tags=["Upload Requirement File"],
-        )
+        # self.router.add_api_route(
+        #     "/sows/{sow_id}/upload",
+        #     self.upload_sow_by_id,
+        #     methods=["POST"],
+        #     response_model=ResponseSOWSingle,
+        #     tags=["Upload Requirement File"],
+        # )
 
     async def get_sows(self):
         async with self.__sow_service() as service:

src/services/_sow_service.py

@@ -34,7 +34,8 @@ class SOWService:
     async def create_sow(self, sow: dict) -> dict:
         try:
             async with self.__repository() as repository:
-                return await repository.create_sow(sow)
+                sow = await repository.create_sow(sow)
+            return await self.generate_sow(sow_id=sow["id"])
         except Exception as e:
             logger.exception(e)
             raise HTTPException(
@@ -47,7 +48,7 @@ class SOWService:
                 result = await repository.update_sow(sow_id=sow_id, sow=sow)
             if not result:
                 raise HTTPException(status_code=404, detail="SOW not found")
-            return result
+            return await self.generate_sow(sow_id=sow_id)
         except HTTPException:
             raise
         except Exception as e:
@@ -107,7 +108,7 @@ class SOWService:
                     id=folder_id, files=files
                 )
 
-            return created_sow if created_sow else sow[0]
+            return await self.generate_sow(sow_id=folder_id)
         except HTTPException:
             raise
         except Exception as e:

src/utils/_sow_client.py

@@ -38,34 +38,47 @@ class SOWClient:
         pass
 
     async def generate_sow(self, sow_id: UUID) -> dict:
+        content = None
+        question_json = None
         async with self._repository() as repository:
             sows = await repository.get_sows(sow_id=sow_id)
         folder_path = os.path.join(os.getcwd(), "temp", str(sow_id))
         if os.path.exists(folder_path):
             content = await self._read_uploaded_files(id=sow_id)
-            sow = sows[0]
-            sow["rfp_content"] = content
-        else:
-            sow = sows[0]
-            sow["rfp_content"] = sow["requirement"]
+        sow = sows[0]
+        sow["requirement"] = content if content else sow["requirement"]
         del sow["id"]
         del sow["created_at"]
         del sow["updated_at"]
         url = os.getenv(
-            "NN_SOW_URL", "https://n8n.srv763317.hstgr.cloud/webhook/generate-sow"
+            "NN_SOW_URL", "https://n8n.srv763317.hstgr.cloud/webhook/sow-generate"
         )
         async with httpx.AsyncClient() as client:
             response = await client.post(
                 url,
                 json=sow,
-                timeout=300,
+                timeout=3000,
             )
             response_json = response.json()
-        generated_sow = response_json["message"]["content"]["statement_of_work"]
+        generated_sow_json = response_json["message"]["content"]
+        if generated_sow_json["is_required_questions"]:
+            question_json = {}
+            for question in generated_sow_json["questions"]:
+                question_json[question] = None
+            question_json = str(question_json)
+        generated_sow = (
+            generated_sow_json["generated_sow"]
+            if generated_sow_json["generated_sow"]
+            else None
+        )
         async with self._repository() as repository:
             updated_sow = await repository.update_sow(
                 sow_id=sow_id,
-                sow={"sow_generated_text": generated_sow},
+                sow={
+                    "sow_generated_text": generated_sow,
+                    "additional_info": question_json,
+                    "requirement": content if content else sow["requirement"],
+                },
             )
         await self._remove_uploaded_files(id=sow_id)
         return updated_sow