fix: test

aurelio-labs · Aug 19, 2024 · e1ef5a0 · e1ef5a0
1 parent f293678
commit e1ef5a0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/semantic_router/utils/function_call.py b/semantic_router/utils/function_call.py
@@ -77,7 +77,11 @@ def to_ollama(self):
                     "type": "object",
                     "properties": {
                         param.name: {
-                            "description": param.description,
+                            "description": (
+                                param.description
+                                if isinstance(param.description, str)
+                                else None
+                            ),
                             "type": self._ollama_type_mapping(param.type),
                         }
                         for param in self.parameters