feat(agents-api): Add embeddings to doc get/list response

agents-api/agents_api/autogen/Docs.py

@@ -57,6 +57,13 @@ class Doc(BaseModel):
     """
     Contents of the document
     """
+    embeddings: Annotated[
+        list[float] | list[list[float]] | None,
+        Field(None, json_schema_extra={"readOnly": True}),
+    ]
+    """
+    Embeddings for the document
+    """
 
 
 class DocOwner(BaseModel):

agents-api/agents_api/models/docs/get_doc.py

@@ -37,6 +37,7 @@
     one=True,
     transform=lambda d: {
         "content": [s[1] for s in sorted(d["snippet_data"], key=lambda x: x[0])],
+        "embeddings": [s[2] for s in sorted(d["snippet_data"], key=lambda x: x[0])],
         **d,
     },
 )
@@ -68,8 +69,9 @@ def get_doc(
                 doc_id,
                 index,
                 content,
+                embedding,
             },
-            snippet_data = [index, content]
+            snippet_data = [index, content, embedding]
 
         ?[
             id,

agents-api/agents_api/models/docs/list_docs.py

@@ -34,6 +34,7 @@
     Doc,
     transform=lambda d: {
         "content": [s[1] for s in sorted(d["snippet_data"], key=lambda x: x[0])],
+        "embeddings": [s[2] for s in sorted(d["snippet_data"], key=lambda x: x[0])],
         **d,
     },
 )
@@ -67,8 +68,9 @@ def list_docs(
                 doc_id: id,
                 index,
                 content,
+                embedding,
             }},
-            snippet_data = [index, content]
+            snippet_data = [index, content, embedding]
 
         ?[
             owner_type,

typespec/docs/models.tsp

@@ -23,6 +23,10 @@ model Doc {
 
     /** Contents of the document */
     content: string | string[];
+
+    /** Embeddings for the document */
+    @visibility("read")
+    embeddings?: float32[] | float32[][];
 }
 
 /** Payload for creating a doc */