feat: bump vectorize-client to 0.4.x and fix tests

Author: nicoloboschi

langchain/pyproject.toml

@@ -10,7 +10,7 @@ authors = [
 requires-python = ">=3.9"
 dependencies = [
     "langchain-core>=0.3.45",
-    "vectorize-client>=0.1.3",
+    "vectorize-client>=0.4.0",
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",

langchain/tests/test_retrievers.py

@@ -68,16 +68,15 @@ def api_client(api_token: str, environment: str) -> Iterator[ApiClient]:
 def pipeline_id(api_client: v.ApiClient, org_id: str) -> Iterator[str]:
     pipelines = v.PipelinesApi(api_client)
 
-    connectors_api = v.ConnectorsApi(api_client)
+    connectors_api = v.SourceConnectorsApi(api_client)
     response = connectors_api.create_source_connector(
         org_id,
-        [
-            v.CreateSourceConnector(
-                name="from api", type=v.SourceConnectorType.FILE_UPLOAD
-            )
-        ],
+            v.CreateSourceConnectorRequest(v.FileUpload(
+        name="from api",
+        type="FILE_UPLOAD"
+    ))
     )
-    source_connector_id = response.connectors[0].id
+    source_connector_id = response.connector.id
     logging.info("Created source connector %s", source_connector_id)
 
     uploads_api = v.UploadsApi(api_client)
@@ -111,13 +110,13 @@ def pipeline_id(api_client: v.ApiClient, org_id: str) -> Iterator[str]:
     else:
         logging.info("Upload successful")
 
-    ai_platforms = connectors_api.get_ai_platform_connectors(org_id)
+    ai_platforms = v.AIPlatformConnectorsApi(api_client).get_ai_platform_connectors(org_id)
     builtin_ai_platform = next(
         c.id for c in ai_platforms.ai_platform_connectors if c.type == "VECTORIZE"
     )
     logging.info("Using AI platform %s", builtin_ai_platform)
 
-    vector_databases = connectors_api.get_destination_connectors(org_id)
+    vector_databases = v.DestinationConnectorsApi(api_client).get_destination_connectors(org_id)
     builtin_vector_db = next(
         c.id for c in vector_databases.destination_connectors if c.type == "VECTORIZE"
     )
@@ -127,24 +126,24 @@ def pipeline_id(api_client: v.ApiClient, org_id: str) -> Iterator[str]:
         org_id,
         v.PipelineConfigurationSchema(
             source_connectors=[
-                v.SourceConnectorSchema(
+                v.PipelineSourceConnectorSchema(
                     id=source_connector_id,
                     type=v.SourceConnectorType.FILE_UPLOAD,
                     config={},
                 )
             ],
-            destination_connector=v.DestinationConnectorSchema(
+            destination_connector=v.PipelineDestinationConnectorSchema(
                 id=builtin_vector_db,
-                type=v.DestinationConnectorType.VECTORIZE,
+                type="VECTORIZE",
                 config={},
             ),
-            ai_platform=v.AIPlatformSchema(
+            ai_platform_connector=v.PipelineAIPlatformConnectorSchema(
                 id=builtin_ai_platform,
-                type=v.AIPlatformType.VECTORIZE,
-                config=v.AIPlatformConfigSchema(),
+                type="VECTORIZE",
+                config={},
             ),
             pipeline_name="Test pipeline",
-            schedule=v.ScheduleSchema(type=v.ScheduleSchemaType.MANUAL),
+            schedule=v.ScheduleSchema(type="manual"),
         ),
     )
     pipeline_id = pipeline_response.data.id