feat(typespec,agents-api): Add mmr_strength to typespec

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Oct 28, 2024 · 010b1d3 · 010b1d3
1 parent 7892cd7
commit 010b1d3
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/agents-api/agents_api/autogen/Docs.py b/agents-api/agents_api/autogen/Docs.py
@@ -19,6 +19,10 @@ class BaseDocSearchRequest(BaseModel):
     The language to be used for text-only search. Support for other languages coming soon.
     """
     metadata_filter: dict[str, float | str | StrictBool | None] = {}
+    mmr_strength: Annotated[float, Field(0, ge=0.0, lt=1.0)]
+    """
+    MMR Strength (mmr_strength = 1 - mmr_lambda)
+    """
 
 
 class CreateDocRequest(BaseModel):

diff --git a/typespec/docs/models.tsp b/typespec/docs/models.tsp
@@ -95,6 +95,11 @@ model BaseDocSearchRequest {
     /** The language to be used for text-only search. Support for other languages coming soon. */
     lang: "en-US" = "en-US";
     metadata_filter: MetadataFilter = #{},
+
+    /** MMR Strength (mmr_strength = 1 - mmr_lambda) */
+    @minValue(0)
+    @maxValueExclusive(1)
+    mmr_strength?: float = 0.0;
 }
 
 model VectorDocSearchRequest extends BaseDocSearchRequest {
@@ -107,6 +112,8 @@ model VectorDocSearchRequest extends BaseDocSearchRequest {
     vector: float[];
 
     text?: never;
+
+    mmr_strength?: never;
 }
 
 model TextOnlyDocSearchRequest extends BaseDocSearchRequest {

diff --git a/typespec/tsp-output/@typespec/openapi3/openapi-1.0.0.yaml b/typespec/tsp-output/@typespec/openapi3/openapi-1.0.0.yaml
@@ -2450,6 +2450,13 @@ components:
               - type: boolean
             nullable: true
           default: {}
+        mmr_strength:
+          type: number
+          minimum: 0
+          maximum: 1
+          exclusiveMaximum: true
+          description: MMR Strength (mmr_strength = 1 - mmr_lambda)
+          default: 0
     Docs.CreateDocRequest:
       type: object
       required: