update

Author: mattgotteiner

app/backend/approaches/approach.py

@@ -4,6 +4,7 @@
 from dataclasses import dataclass
 from typing import Any, Callable, Optional, TypedDict, Union, cast
 from urllib.parse import urljoin
+import json
 
 import aiohttp
 from azure.search.documents.agent.aio import KnowledgeAgentRetrievalClient
@@ -297,19 +298,26 @@ async def run_agentic_retrieval(
         )
 
         results = []
-        if response and response.references:
-            for reference in response.references:
-                if isinstance(reference, KnowledgeAgentAzureSearchDocReference) and reference.source_data:
-                    results.append(
-                        Document(
-                            id=reference.doc_key,
-                            content=reference.source_data["content"],
-                            sourcepage=reference.source_data["sourcepage"],
-                            search_agent_query=activity_mapping[reference.activity_source],
-                        )
-                    )
-                if top and len(results) == top:
-                    break
+        grounding_text = response.response[0].content[0].text
+        if response and grounding_text:
+            grounding_response = json.loads(grounding_text)
+            for grounding_reference in grounding_response:
+                ref_id = grounding_reference.get("ref_id")
+                if ref_id is not None:
+                    ref_id = str(ref_id)
+                    for reference in response.references:
+                        if isinstance(reference, KnowledgeAgentAzureSearchDocReference) and reference.id == ref_id and reference.source_data:
+                            results.append(
+                                Document(
+                                    id=reference.doc_key,
+                                    content=reference.source_data["content"],
+                                    sourcepage=reference.source_data["sourcepage"],
+                                    search_agent_query=activity_mapping[reference.activity_source],
+                                )
+                            )
+                            break
+                    if top and len(results) == top:
+                        break
 
         return response, results
 

app/frontend/src/pages/ask/Ask.tsx

@@ -102,10 +102,6 @@ export function Component(): JSX.Element {
             setShowSpeechOutputAzure(config.showSpeechOutputAzure);
             setShowAgenticRetrievalOption(config.showAgenticRetrievalOption);
             setUseAgenticRetrieval(config.showAgenticRetrievalOption);
-            if (config.showAgenticRetrievalOption) {
-                setRetrieveCount(10);
-                setMinimumRerankerScore(2.1);
-            }
         });
     };
 

app/frontend/src/pages/chat/Chat.tsx

@@ -135,10 +135,6 @@ const Chat = () => {
             setShowChatHistoryCosmos(config.showChatHistoryCosmos);
             setShowAgenticRetrievalOption(config.showAgenticRetrievalOption);
             setUseAgenticRetrieval(config.showAgenticRetrievalOption);
-            if (config.showAgenticRetrievalOption) {
-                setRetrieveCount(10);
-                setMinimumRerankerScore(2.1);
-            }
         });
     };