fix: one-pipe manifold

This commit is contained in:
Justin Hayes 2024-07-26 10:54:43 -04:00 committed by GitHub
parent 984d5b9f58
commit a26527679b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -50,15 +50,16 @@ class Pipeline:
)
def __init__(self):
self.type = "manifold"
self.valves = self.Valves()
self.id = f"routellm-{self.valves.ROUTELLM_SUFFIX.lower()}"
self.name = f"RouteLLM/{self.valves.ROUTELLM_SUFFIX}"
self.id = "routellm"
self.name = f"RouteLLM/"
self.controller = None
self._initialize_controller()
def pipelines(self) -> List[dict]:
return [{"id": f"routellm.{self.valves.ROUTELLM_ROUTER}", "name": f"RouteLLM/{self.valves.ROUTELLM_ROUTER}"}]
return [{"id": f"{self.valves.ROUTELLM_SUFFIX.lower()}", "name": f"{self.valves.ROUTELLM_SUFFIX}"}]
async def on_startup(self):
logging.info(f"on_startup: {__name__}")