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

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__}")