Merge pull request #49 from Serverless-Devs/add-adb-retrieve

feat(api): add loader_metadata to ADBDataAPI response

Author: OhYee

agentrun/knowledgebase/api/__data_async_template.py

@@ -467,6 +467,11 @@ def _parse_query_content_response(
                     "metadata": (
                         match.metadata if hasattr(match, "metadata") else None
                     ),
+                    "loader_metadata": (
+                        match.loader_metadata
+                        if hasattr(match, "loader_metadata")
+                        else None
+                    ),
                     "rerank_score": (
                         match.rerank_score
                         if hasattr(match, "rerank_score")

agentrun/knowledgebase/api/data.py

@@ -677,6 +677,11 @@ def _parse_query_content_response(
                     "metadata": (
                         match.metadata if hasattr(match, "metadata") else None
                     ),
+                    "loader_metadata": (
+                        match.loader_metadata
+                        if hasattr(match, "loader_metadata")
+                        else None
+                    ),
                     "rerank_score": (
                         match.rerank_score
                         if hasattr(match, "rerank_score")

pyproject.toml

@@ -42,6 +42,7 @@ agentscope = [
 
 crewai = [
     "crewai>=1.6.0",
+    "onnxruntime==1.23.2; python_version == '3.10'",
 ]
 
 pydantic-ai = [