From 6d92b38167dbd9a7555f4d72701a384a8b73bbdc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Boschi?= Date: Mon, 24 Jun 2024 09:41:08 +0200 Subject: [PATCH] fixes and refactory --- .../base/langflow/components/embeddings/AstraVectorize.py | 4 ++-- src/backend/base/langflow/services/telemetry/service.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/embeddings/AstraVectorize.py b/src/backend/base/langflow/components/embeddings/AstraVectorize.py index 16d3cf316..118a28442 100644 --- a/src/backend/base/langflow/components/embeddings/AstraVectorize.py +++ b/src/backend/base/langflow/components/embeddings/AstraVectorize.py @@ -42,10 +42,10 @@ class AstraVectorize(Component): ), ] outputs = [ - Output(display_name="Vectorize", name="config", method="build", types=["dict"]), + Output(display_name="Vectorize", name="config", method="build_options", types=["dict"]), ] - def build(self) -> dict[str, Any]: + def build_options(self) -> dict[str, Any]: return { "collection_vector_service_options": { "provider": self.provider, diff --git a/src/backend/base/langflow/services/telemetry/service.py b/src/backend/base/langflow/services/telemetry/service.py index 15ccacf69..76e82835c 100644 --- a/src/backend/base/langflow/services/telemetry/service.py +++ b/src/backend/base/langflow/services/telemetry/service.py @@ -29,7 +29,7 @@ class TelemetryService(Service): super().__init__() self.settings_service = settings_service self.base_url = settings_service.settings.telemetry_base_url - self.telemetry_queue = asyncio.Queue() + self.telemetry_queue: asyncio.Queue = asyncio.Queue() self.client = httpx.AsyncClient(timeout=None) self.running = False self.package = get_version_info()["package"]