bug 550 : make sure the model_name with the ollama list (stitionai#551)

chisun-joung · web-flow · commit 4a62feb782ac · 2024-05-15T21:39:29.000+05:30
close stitionai#550
diff --git a/src/llm/llm.py b/src/llm/llm.py
@@ -60,7 +60,7 @@ def __init__(self, model_id: str = None):
             "OLLAMA": []
         }
         if ollama.client:
-            self.models["OLLAMA"] = [(model["name"].split(":")[0], model["name"]) for model in ollama.models]
+            self.models["OLLAMA"] = [(model["name"], model["name"]) for model in ollama.models]
 
     def list_models(self) -> dict:
         return self.models

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ def __init__(self, model_id: str = None):`
`60`	`60`	`"OLLAMA": []`
`61`	`61`	`}`
`62`	`62`	`if ollama.client:`
`63`		`- self.models["OLLAMA"] = [(model["name"].split(":")[0], model["name"]) for model in ollama.models]`
	`63`	`+ self.models["OLLAMA"] = [(model["name"], model["name"]) for model in ollama.models]`
`64`	`64`
`65`	`65`	`def list_models(self) -> dict:`
`66`	`66`	`return self.models`