|
4 | 4 |
|
5 | 5 | import httpx |
6 | 6 |
|
7 | | -from ..types import Query, search_execute_params, search_memories_params, search_documents_params |
| 7 | +from ..types import search_execute_params, search_memories_params, search_documents_params |
8 | 8 | from .._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given |
9 | 9 | from .._utils import maybe_transform, async_maybe_transform |
10 | 10 | from .._compat import cached_property |
|
16 | 16 | async_to_streamed_response_wrapper, |
17 | 17 | ) |
18 | 18 | from .._base_client import make_request_options |
19 | | -from ..types.query_param import QueryParam |
20 | 19 | from ..types.search_execute_response import SearchExecuteResponse |
21 | 20 | from ..types.search_memories_response import SearchMemoriesResponse |
22 | 21 | from ..types.search_documents_response import SearchDocumentsResponse |
@@ -53,7 +52,7 @@ def documents( |
53 | 52 | container_tags: SequenceNotStr[str] | Omit = omit, |
54 | 53 | doc_id: str | Omit = omit, |
55 | 54 | document_threshold: float | Omit = omit, |
56 | | - filters: QueryParam | Omit = omit, |
| 55 | + filters: search_documents_params.Filters | Omit = omit, |
57 | 56 | include_full_docs: bool | Omit = omit, |
58 | 57 | include_summary: bool | Omit = omit, |
59 | 58 | limit: int | Omit = omit, |
@@ -151,7 +150,7 @@ def execute( |
151 | 150 | container_tags: SequenceNotStr[str] | Omit = omit, |
152 | 151 | doc_id: str | Omit = omit, |
153 | 152 | document_threshold: float | Omit = omit, |
154 | | - filters: QueryParam | Omit = omit, |
| 153 | + filters: search_execute_params.Filters | Omit = omit, |
155 | 154 | include_full_docs: bool | Omit = omit, |
156 | 155 | include_summary: bool | Omit = omit, |
157 | 156 | limit: int | Omit = omit, |
@@ -245,7 +244,7 @@ def memories( |
245 | 244 | *, |
246 | 245 | q: str, |
247 | 246 | container_tag: str | Omit = omit, |
248 | | - filters: QueryParam | Omit = omit, |
| 247 | + filters: search_memories_params.Filters | Omit = omit, |
249 | 248 | include: search_memories_params.Include | Omit = omit, |
250 | 249 | limit: int | Omit = omit, |
251 | 250 | rerank: bool | Omit = omit, |
@@ -340,7 +339,7 @@ async def documents( |
340 | 339 | container_tags: SequenceNotStr[str] | Omit = omit, |
341 | 340 | doc_id: str | Omit = omit, |
342 | 341 | document_threshold: float | Omit = omit, |
343 | | - filters: QueryParam | Omit = omit, |
| 342 | + filters: search_documents_params.Filters | Omit = omit, |
344 | 343 | include_full_docs: bool | Omit = omit, |
345 | 344 | include_summary: bool | Omit = omit, |
346 | 345 | limit: int | Omit = omit, |
@@ -438,7 +437,7 @@ async def execute( |
438 | 437 | container_tags: SequenceNotStr[str] | Omit = omit, |
439 | 438 | doc_id: str | Omit = omit, |
440 | 439 | document_threshold: float | Omit = omit, |
441 | | - filters: QueryParam | Omit = omit, |
| 440 | + filters: search_execute_params.Filters | Omit = omit, |
442 | 441 | include_full_docs: bool | Omit = omit, |
443 | 442 | include_summary: bool | Omit = omit, |
444 | 443 | limit: int | Omit = omit, |
@@ -532,7 +531,7 @@ async def memories( |
532 | 531 | *, |
533 | 532 | q: str, |
534 | 533 | container_tag: str | Omit = omit, |
535 | | - filters: QueryParam | Omit = omit, |
| 534 | + filters: search_memories_params.Filters | Omit = omit, |
536 | 535 | include: search_memories_params.Include | Omit = omit, |
537 | 536 | limit: int | Omit = omit, |
538 | 537 | rerank: bool | Omit = omit, |
|
0 commit comments