Fix references extraction

app.py

@@ -111,6 +111,7 @@ async def categorize_reqs(params: ReqGroupingRequest) -> ReqGroupingResponse:
 
 # ========================================================= Solution Endpoints ===========================================================
 
+
 @solution_router.post("/search_solutions_gemini", response_model=SolutionSearchResponse)
 async def search_solutions(params: ReqGroupingResponse) -> SolutionSearchResponse:
     """Searches solutions using Gemini and grounded on google search"""
@@ -144,8 +145,8 @@ async def search_solutions(params: ReqGroupingResponse) -> SolutionSearchRespons
         # ======================== build the final solution object ================================
 
         # extract the source metadata from the search items
-        sources_metadata = [
-            f'{a["web"]["title"]} - {a["web"]["uri"]}' for a in req_completion["vertex_ai_grounding_metadata"][0]['groundingChunks']]
+        sources_metadata = [{"name": a["web"]["title"], "url": a["web"]["uri"]}
+                            for a in req_completion["vertex_ai_grounding_metadata"][0]['groundingChunks']]
 
         final_sol = SolutionModel(
             Context="",
@@ -166,6 +167,7 @@ async def search_solutions(params: ReqGroupingResponse) -> SolutionSearchRespons
 
     return SolutionSearchResponse(solutions=final_solutions)
 
+
 @solution_router.post("/criticize_solution", response_model=CritiqueResponse)
 async def criticize_solution(params: CriticizeSolutionsRequest) -> CritiqueResponse:
     """Criticize the challenges, weaknesses and limitations of the provided solutions."""

schemas.py

@@ -29,7 +29,7 @@ class SolutionModel(BaseModel):
                                      description="Description of the problem being solved.")
     Solution_Description: str = Field(..., alias="Solution Description",
                                       description="Detailed description of the solution.")
-    References: list[str] = Field(
+    References: list[dict] = Field(
         ..., description="References to documents used for the solution.")
     Category_Id: int = Field(
         ..., description="ID of the requirements category the solution is based on")