fix search

Author: ssmith-pc

pinecone/repository_data/repository.py

@@ -250,10 +250,10 @@ def delete(self, namespace: str, document_id: str, **kwargs):
     def search(
         self,
         namespace: str,
-        query_text: str,
+        query_str: str,
         top_k: Optional[int] = None,
         filter: Optional[Dict[str, Any]] = None,
     ) -> SearchDocumentsResponse:
         return self._search_client.search(
-            namespace=namespace, query_text=query_text, top_k=top_k, filter=filter
+            namespace=namespace, query_str=query_str, top_k=top_k, filter=filter
         )

pinecone/repository_data/repository_search.py

@@ -106,15 +106,15 @@ def close(self):
     def search(
         self,
         namespace: str,
-        query_text: str,
+        query_str: str,
         top_k: Optional[int] = 10,
         filter: Optional[Dict[str, Any]] = None,
     ) -> SearchDocumentsResponse:
         if namespace is None:
             raise Exception("Namespace is required when searching documents")
 
         request = RepositoryRequestFactory.search_request(
-            namespace=namespace, query_text=query_text, top_k=top_k, filter=filter
+            query_str=query_str, top_k=top_k, filter=filter
         )
 
-        return self._repo_api.search_records_namespace(namespace, request)
+        return self._repo_api.search_documents(namespace, request)

pinecone/repository_data/request_factory.py

@@ -21,11 +21,11 @@ def non_openapi_kwargs(kwargs: Dict[str, Any]) -> Dict[str, Any]:
 class RepositoryRequestFactory:
     @staticmethod
     def search_request(
-        query_text: str,
+        query_str: str,
         top_k: int,
         filter: Optional[Dict[str, Any]] = None,
         # TODO, add options
     ) -> SearchDocuments:
         return SearchDocuments(
-            query=QueryModel(inputs=QueryInputModel(text=query_text), top_k=top_k, filter=filter)
+            query=QueryModel(inputs=QueryInputModel(text=query_str), top_k=top_k, filter=filter)
         )