@@ -60,9 +60,15 @@ def insert(
6060 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
6161 ) -> None :
6262 """
63- Index documents so they can be used by the RAG system
63+ Index documents so they can be used by the RAG system.
6464
6565 Args:
66+ chunk_size_in_tokens: (Optional) Size in tokens for document chunking during indexing
67+
68+ documents: List of documents to index in the RAG system
69+
70+ vector_db_id: ID of the vector database to store the document embeddings
71+
6672 extra_headers: Send extra headers
6773
6874 extra_query: Add additional query parameters to the request
@@ -102,12 +108,14 @@ def query(
102108 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
103109 ) -> QueryResult :
104110 """
105- Query the RAG system for context; typically invoked by the agent
111+ Query the RAG system for context; typically invoked by the agent.
106112
107113 Args:
108- content: A image content item
114+ content: The query content to search for in the indexed documents
115+
116+ vector_db_ids: List of vector database IDs to search within
109117
110- query_config: Configuration for the RAG query generation.
118+ query_config: (Optional) Configuration parameters for the query operation
111119
112120 extra_headers: Send extra headers
113121
@@ -168,9 +176,15 @@ async def insert(
168176 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
169177 ) -> None :
170178 """
171- Index documents so they can be used by the RAG system
179+ Index documents so they can be used by the RAG system.
172180
173181 Args:
182+ chunk_size_in_tokens: (Optional) Size in tokens for document chunking during indexing
183+
184+ documents: List of documents to index in the RAG system
185+
186+ vector_db_id: ID of the vector database to store the document embeddings
187+
174188 extra_headers: Send extra headers
175189
176190 extra_query: Add additional query parameters to the request
@@ -210,12 +224,14 @@ async def query(
210224 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
211225 ) -> QueryResult :
212226 """
213- Query the RAG system for context; typically invoked by the agent
227+ Query the RAG system for context; typically invoked by the agent.
214228
215229 Args:
216- content: A image content item
230+ content: The query content to search for in the indexed documents
231+
232+ vector_db_ids: List of vector database IDs to search within
217233
218- query_config: Configuration for the RAG query generation.
234+ query_config: (Optional) Configuration parameters for the query operation
219235
220236 extra_headers: Send extra headers
221237
0 commit comments