wip

app.py

@@ -1,4 +1,5 @@
 import asyncio
+import json
 import logging
 import os
 import sys
@@ -330,8 +331,13 @@ async def search_solutions_if(req: SolutionSearchV2Request) -> SolutionSearchRes
         fmt_model = InsightFinderConstraintsList.model_validate_json(
             fmt_completion.choices[0].message.content)
 
+        out = {'constraints': {
+            cons.title: cons.description for cons in fmt_model.constraints}}
+        # logging.info(out)
+
         # fetch technologies from insight finder
-        technologies_req = await http_client.post(INSIGHT_FINDER_BASE_URL + "process-constraints", content=fmt_model.model_dump_json())
+        # translate from a structured output to a dict for insights finder
+        technologies_req = await http_client.post(INSIGHT_FINDER_BASE_URL + "process-constraints", content=json.dumps(out))
         technologies = TechnologyData.model_validate(technologies_req.json())
 
         # =============================================================== synthesize solution using LLM =========================================
@@ -364,7 +370,7 @@ async def search_solutions_if(req: SolutionSearchV2Request) -> SolutionSearchRes
 
         return final_solution
 
-    tasks = await asyncio.gather(*[_search_solution_inner(cat) for cat in req.categories])
+    tasks = await asyncio.gather(*[_search_solution_inner(cat) for cat in req.categories], return_exceptions=True)
     final_solutions = [sol for sol in tasks if not isinstance(sol, Exception)]
 
     return SolutionSearchResponse(solutions=final_solutions)

prompts/if/format_requirements.txt

@@ -1,8 +1,8 @@
 <role>You are an expert system designer</role>
 <task>
-Your task is to transform a category of requirements into a JSON object where the object key is a short title (with spaces) for the requirement and the value is the requirement.
-Give the the most important constraints that must be solved to address the general problem of the category.
-Give at most the 10 most important constraints.
+Your task is to extract from a category of requirements the top most technical constraints that must be solved.
+Give the most important constraints that must be solved to address the general problem of the category.
+Give at most the 10 constraints.
 </task>
 
 <requirements>

schemas.py

@@ -138,9 +138,17 @@ class _RefinedSolutionModel(BaseModel):
 
 # ================================================================= search solutions using insight finder endpoints
 
+# Helpers to extract constraints for Insights Finder
+
+class InsightFinderConstraintItem(BaseModel):
+    title: str
+    description: str
+
+
 class InsightFinderConstraintsList(BaseModel):
-    constraints: dict = Field(
-        description="A dict of constraints where the key is a short title for the constraint and the value the whole constraint description")
+    constraints: list[InsightFinderConstraintItem]
+
+# =================================================
 
 
 class Technology(BaseModel):
@@ -152,9 +160,6 @@ class Technology(BaseModel):
     limitations: str
     id: int
 
-# This schema defines the root structure of the JSON
-
-
 class TechnologyData(BaseModel):
     """Represents the top-level object containing a list of technologies."""
     technologies: List[Technology]