diff --git a/.stats.yml b/.stats.yml index aa6dcc4c..9688162b 100644 --- a/.stats.yml +++ b/.stats.yml @@ -1,2 +1,2 @@ configured_endpoints: 53 -openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/julep-ai-inc-dash%2Fjulep-dc6080a0ac01dc48d514645804c7dc3b13876aac720e09b1f3654ed301daeff1.yml +openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/julep-ai-inc-dash%2Fjulep-23aec0f2b6f0820456a72a3cd0083f6c8b766f3af3fcc21f0f7e0c6fd34bbdf2.yml diff --git a/src/julep/types/session.py b/src/julep/types/session.py index bbd680d6..a39566f0 100644 --- a/src/julep/types/session.py +++ b/src/julep/types/session.py @@ -10,12 +10,20 @@ class RecallOptions(BaseModel): + alpha: Optional[float] = None + confidence: Optional[float] = None - hybrid_alpha: Optional[float] = None + lang: Optional[Literal["en-US"]] = None + + limit: Optional[int] = None max_query_length: Optional[int] = None + metadata_filter: Optional[object] = None + + mmr_strength: Optional[float] = None + mode: Optional[Literal["hybrid", "vector", "text"]] = None num_search_messages: Optional[int] = None diff --git a/src/julep/types/session_create_or_update_params.py b/src/julep/types/session_create_or_update_params.py index 47451f3f..19c96911 100644 --- a/src/julep/types/session_create_or_update_params.py +++ b/src/julep/types/session_create_or_update_params.py @@ -33,12 +33,20 @@ class SessionCreateOrUpdateParams(TypedDict, total=False): class RecallOptions(TypedDict, total=False): + alpha: float + confidence: float - hybrid_alpha: float + lang: Literal["en-US"] + + limit: int max_query_length: int + metadata_filter: object + + mmr_strength: float + mode: Literal["hybrid", "vector", "text"] num_search_messages: int diff --git a/src/julep/types/session_create_params.py b/src/julep/types/session_create_params.py index 83af632d..7cbf70ae 100644 --- a/src/julep/types/session_create_params.py +++ b/src/julep/types/session_create_params.py @@ -33,12 +33,20 @@ class SessionCreateParams(TypedDict, total=False): class RecallOptions(TypedDict, total=False): + alpha: float + confidence: float - hybrid_alpha: float + lang: Literal["en-US"] + + limit: int max_query_length: int + metadata_filter: object + + mmr_strength: float + mode: Literal["hybrid", "vector", "text"] num_search_messages: int diff --git a/src/julep/types/session_patch_params.py b/src/julep/types/session_patch_params.py index b73e6f4b..d15a4c5f 100644 --- a/src/julep/types/session_patch_params.py +++ b/src/julep/types/session_patch_params.py @@ -25,12 +25,20 @@ class SessionPatchParams(TypedDict, total=False): class RecallOptions(TypedDict, total=False): + alpha: float + confidence: float - hybrid_alpha: float + lang: Literal["en-US"] + + limit: int max_query_length: int + metadata_filter: object + + mmr_strength: float + mode: Literal["hybrid", "vector", "text"] num_search_messages: int diff --git a/src/julep/types/session_update_params.py b/src/julep/types/session_update_params.py index 42634722..26160a4f 100644 --- a/src/julep/types/session_update_params.py +++ b/src/julep/types/session_update_params.py @@ -25,12 +25,20 @@ class SessionUpdateParams(TypedDict, total=False): class RecallOptions(TypedDict, total=False): + alpha: float + confidence: float - hybrid_alpha: float + lang: Literal["en-US"] + + limit: int max_query_length: int + metadata_filter: object + + mmr_strength: float + mode: Literal["hybrid", "vector", "text"] num_search_messages: int diff --git a/tests/api_resources/test_sessions.py b/tests/api_resources/test_sessions.py index f9277c9d..2e0c7598 100644 --- a/tests/api_resources/test_sessions.py +++ b/tests/api_resources/test_sessions.py @@ -37,9 +37,13 @@ def test_method_create_with_all_params(self, client: Julep) -> None: context_overflow="truncate", metadata={}, recall_options={ + "alpha": 0, "confidence": 0, - "hybrid_alpha": 0, + "lang": "en-US", + "limit": 1, "max_query_length": 0, + "metadata_filter": {}, + "mmr_strength": 0, "mode": "hybrid", "num_search_messages": 0, }, @@ -86,9 +90,13 @@ def test_method_update_with_all_params(self, client: Julep) -> None: context_overflow="truncate", metadata={}, recall_options={ + "alpha": 0, "confidence": 0, - "hybrid_alpha": 0, + "lang": "en-US", + "limit": 1, "max_query_length": 0, + "metadata_filter": {}, + "mmr_strength": 0, "mode": "hybrid", "num_search_messages": 0, }, @@ -381,9 +389,13 @@ def test_method_create_or_update_with_all_params(self, client: Julep) -> None: context_overflow="truncate", metadata={}, recall_options={ + "alpha": 0, "confidence": 0, - "hybrid_alpha": 0, + "lang": "en-US", + "limit": 1, "max_query_length": 0, + "metadata_filter": {}, + "mmr_strength": 0, "mode": "hybrid", "num_search_messages": 0, }, @@ -517,9 +529,13 @@ def test_method_patch_with_all_params(self, client: Julep) -> None: context_overflow="truncate", metadata={}, recall_options={ + "alpha": 0, "confidence": 0, - "hybrid_alpha": 0, + "lang": "en-US", + "limit": 1, "max_query_length": 0, + "metadata_filter": {}, + "mmr_strength": 0, "mode": "hybrid", "num_search_messages": 0, }, @@ -578,9 +594,13 @@ async def test_method_create_with_all_params(self, async_client: AsyncJulep) -> context_overflow="truncate", metadata={}, recall_options={ + "alpha": 0, "confidence": 0, - "hybrid_alpha": 0, + "lang": "en-US", + "limit": 1, "max_query_length": 0, + "metadata_filter": {}, + "mmr_strength": 0, "mode": "hybrid", "num_search_messages": 0, }, @@ -627,9 +647,13 @@ async def test_method_update_with_all_params(self, async_client: AsyncJulep) -> context_overflow="truncate", metadata={}, recall_options={ + "alpha": 0, "confidence": 0, - "hybrid_alpha": 0, + "lang": "en-US", + "limit": 1, "max_query_length": 0, + "metadata_filter": {}, + "mmr_strength": 0, "mode": "hybrid", "num_search_messages": 0, }, @@ -922,9 +946,13 @@ async def test_method_create_or_update_with_all_params(self, async_client: Async context_overflow="truncate", metadata={}, recall_options={ + "alpha": 0, "confidence": 0, - "hybrid_alpha": 0, + "lang": "en-US", + "limit": 1, "max_query_length": 0, + "metadata_filter": {}, + "mmr_strength": 0, "mode": "hybrid", "num_search_messages": 0, }, @@ -1058,9 +1086,13 @@ async def test_method_patch_with_all_params(self, async_client: AsyncJulep) -> N context_overflow="truncate", metadata={}, recall_options={ + "alpha": 0, "confidence": 0, - "hybrid_alpha": 0, + "lang": "en-US", + "limit": 1, "max_query_length": 0, + "metadata_filter": {}, + "mmr_strength": 0, "mode": "hybrid", "num_search_messages": 0, },