daarion-admin/microdao-daarion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: register vision_encoder from orchestrator_providers config

Browse Source
This commit is contained in:
Apple
2025-11-17 06:14:08 -08:00
parent 212eb0a5a0
commit b521c9cfc5
1 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
providers/registry.py
View File

@@ -81,9 +81,9 @@ def build_provider_registry(config: RouterConfig) -> Dict[str, Provider]:
# Build Orchestrator providers # Build Orchestrator providers
for orch_id, orch_config in config.orchestrator_providers.items(): for orch_id, orch_config in config.orchestrator_providers.items():
provider_id = f"orchestrator_{orch_id}"
if orch_config.get("type") == "orchestrator": if orch_config.get("type") == "orchestrator":
provider_id = f"orchestrator_{orch_id}"
provider = CrewAIProvider( provider = CrewAIProvider(
provider_id=provider_id, provider_id=provider_id,
base_url=orch_config["base_url"], base_url=orch_config["base_url"],
@@ -93,22 +93,23 @@ def build_provider_registry(config: RouterConfig) -> Dict[str, Provider]:
registry[provider_id] = provider registry[provider_id] = provider
base_url = orch_config.get("base_url", "N/A") base_url = orch_config.get("base_url", "N/A")
logger.info(f" + {provider_id}: Orchestrator @ {base_url}") 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: else:
orch_type = orch_config.get("type", "N/A") orch_type = orch_config.get("type", "N/A")
logger.warning(f"Unknown orchestrator type: {orch_type}") 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") logger.info(f"Provider registry built: {len(registry)} providers")
return registry return registry