From b882dd79ff9b6a21f362a8a5818946573ada1134 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sun, 2 Jun 2024 17:00:44 -0700 Subject: [PATCH] fix: litellm pipelines --- examples/pipelines/providers/litellm_manifold_pipeline.py | 2 +- .../pipelines/providers/litellm_subprocess_manifold_pipeline.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/pipelines/providers/litellm_manifold_pipeline.py b/examples/pipelines/providers/litellm_manifold_pipeline.py index 40c891cb..0b6771fa 100644 --- a/examples/pipelines/providers/litellm_manifold_pipeline.py +++ b/examples/pipelines/providers/litellm_manifold_pipeline.py @@ -70,7 +70,7 @@ def get_litellm_models(self): print(f"Error: {e}") return [ { - "id": self.id, + "id": "error", "name": "Could not fetch models from LiteLLM, please update the URL in the valves.", }, ] diff --git a/examples/pipelines/providers/litellm_subprocess_manifold_pipeline.py b/examples/pipelines/providers/litellm_subprocess_manifold_pipeline.py index 4fab306f..4213c36a 100644 --- a/examples/pipelines/providers/litellm_subprocess_manifold_pipeline.py +++ b/examples/pipelines/providers/litellm_subprocess_manifold_pipeline.py @@ -173,7 +173,7 @@ def get_litellm_models(self): print(f"Error: {e}") return [ { - "id": self.id, + "id": "error", "name": "Could not fetch models from LiteLLM, please update the URL in the valves.", }, ]