From b521c9cfc5cc12ab4e44726f00c6018b1cd3ae2d Mon Sep 17 00:00:00 2001 From: Apple Date: Mon, 17 Nov 2025 06:14:08 -0800 Subject: [PATCH] fix: register vision_encoder from orchestrator_providers config --- providers/registry.py | 27 ++++++++++++++------------- 1 file changed, 14 insertions(+), 13 deletions(-) diff --git a/providers/registry.py b/providers/registry.py index 320d07b7..792bd8a4 100644 --- a/providers/registry.py +++ b/providers/registry.py @@ -81,9 +81,9 @@ def build_provider_registry(config: RouterConfig) -> Dict[str, Provider]: # Build Orchestrator providers for orch_id, orch_config in config.orchestrator_providers.items(): - provider_id = f"orchestrator_{orch_id}" if orch_config.get("type") == "orchestrator": + provider_id = f"orchestrator_{orch_id}" provider = CrewAIProvider( provider_id=provider_id, base_url=orch_config["base_url"], @@ -93,22 +93,23 @@ def build_provider_registry(config: RouterConfig) -> Dict[str, Provider]: registry[provider_id] = provider base_url = orch_config.get("base_url", "N/A") logger.info(f" + {provider_id}: Orchestrator @ {base_url}") + + elif orch_config.get("type") == "vision": + provider_id = orch_id # Use orch_id directly (e.g. "vision_encoder") + provider = VisionEncoderProvider( + provider_id=provider_id, + base_url=orch_config["base_url"], + timeout=orch_config.get("timeout_ms", 30000) // 1000 + ) + + registry[provider_id] = provider + base_url = orch_config.get("base_url", "N/A") + logger.info(f" + {provider_id}: VisionEncoder @ {base_url}") + else: orch_type = orch_config.get("type", "N/A") logger.warning(f"Unknown orchestrator type: {orch_type}") - # Build Vision Encoder provider - vision_encoder_url = os.getenv("VISION_ENCODER_URL", "http://vision-encoder:8001") - if vision_encoder_url: - provider_id = "vision_encoder" - provider = VisionEncoderProvider( - provider_id=provider_id, - base_url=vision_encoder_url, - timeout=60 - ) - registry[provider_id] = provider - logger.info(f" + {provider_id}: VisionEncoder @ {vision_encoder_url}") - logger.info(f"Provider registry built: {len(registry)} providers") return registry