adding more telemetry for timing cosmos calls

Author: sethjuarez

api/voice/__init__.py

@@ -7,7 +7,7 @@
 from api.voice.common import (
     get_cosmos_container,
     load_prompty_config,
-    seed_configurations,
+    query_configurations,
 )
 
 
@@ -29,23 +29,7 @@ class Config(BaseModel):
 
 @router.get("/")
 async def get_configurations():
-    async with get_cosmos_container() as container:
-        items = container.read_all_items()
-        configurations: list[Configuration] = []
-        async for item in items:
-            configurations.append(
-                Configuration(
-                    id=item["id"],
-                    name=item["name"],
-                    default=item["default"] if "default" in item else False,
-                    content=item["content"],
-                    tools=item["tools"] if "tools" in item else [],
-                )
-            )
-
-        if len(configurations) == 0:
-            configurations = await seed_configurations(container)
-
+    configurations = await query_configurations()
     return configurations
 
 

api/voice/common.py

@@ -23,6 +23,7 @@
 CONTAINER_NAME = "VoiceConfigurations"
 
 
+@trace
 async def seed_configurations(container: ContainerProxy) -> list[Configuration]:
     configs = []
     # Load default configuration from file
@@ -132,6 +133,28 @@ async def get_cosmos_container():
         await client.close()
 
 
+@trace
+async def query_configurations() -> list[Configuration]:
+    async with get_cosmos_container() as container:
+        items = container.read_all_items()
+        configurations: list[Configuration] = []
+        async for item in items:
+            configurations.append(
+                Configuration(
+                    id=item["id"],
+                    name=item["name"],
+                    default=item["default"] if "default" in item else False,
+                    content=item["content"],
+                    tools=item["tools"] if "tools" in item else [],
+                )
+            )
+
+        if len(configurations) == 0:
+            configurations = await seed_configurations(container)
+            
+    return configurations
+
+
 @trace
 async def get_default_configuration() -> Union[Configuration, None]:
     async with get_cosmos_container() as container:
@@ -163,6 +186,7 @@ def convert_function_params(params: list[dict]) -> dict:
         "required": [p["name"] for p in params if p["required"]],
     }
 
+
 @trace
 async def get_default_configuration_data(**args) -> Union[DefaultConfiguration, None]:
     config = await get_default_configuration()