From c1979b41dda817ac39c9a46905710c74f507c155 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sun, 10 Nov 2024 18:54:37 +0100
Subject: [PATCH] fix: include metadata and func schemas in get_utterances
---
semantic_router/index/base.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/semantic_router/index/base.py b/semantic_router/index/base.py
index da4fba54..37e7be55 100644
--- a/semantic_router/index/base.py
+++ b/semantic_router/index/base.py
@@ -50,7 +50,15 @@ class BaseIndex(BaseModel):
_, metadata = self._get_all(include_metadata=True)
route_tuples: List[
Tuple[str, str, Optional[Dict[str, Any]], Dict[str, Any]]
- ] = [(x["sr_route"], x["sr_utterance"], None, {}) for x in metadata]
+ ] = [
+ (
+ x["sr_route"],
+ x["sr_utterance"],
+ x.get("sr_function_schema", None),
+ x.get("sr_metadata", {}),
+ )
+ for x in metadata
+ ]
return route_tuples
def get_routes(self) -> List[Route]:
--
GitLab