OpenBMB/ChatDev
1
1
Fork 0
mirror of https://github.com/OpenBMB/ChatDev.git synced 2024-12-25 04:44:50 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed: [한병찬] fixed error by base url

Browse Source
This commit is contained in:
onebottlekick 2024-09-26 10:37:55 +09:00
parent 13a9330d4e
commit 602a5001d0
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
camel/model_backend.py
View File

@ -38,10 +38,7 @@ if "OPENAI_API_KEY" in os.environ:
if 'BASE_URL' in os.environ:
BASE_URL = os.environ['BASE_URL']
else:
if "MODEL_NAME" in os.environ:
BASE_URL = os.environ['BASE_URL']
else:
BASE_URL = None
BASE_URL = None
def convert_ollama_to_openai(output):
@ -302,7 +299,7 @@ class StubModel(ModelBackend):
class Ollama(ModelBackend):
r"""OLLAMA API in a unified ModelBackend interface."""
BASE_URL = "http://127.0.0.1:11434" if BASE_URL is None else BASE_URL
def __init__(self, model_type: ModelType, model_config_dict: Dict) -> None:
super().__init__()
self.model_type = model_type
@ -317,7 +314,7 @@ class Ollama(ModelBackend):
_model_name = os.environ["MODEL_NAME"]
kwargs["model"] = _model_name
url = f"{BASE_URL}/api/chat"
url = f"{self.BASE_URL}/api/chat"
data = {"model": _model_name, "messages": kwargs["messages"], "stream": False}
response = requests.post(url, json=data).json()
response = convert_ollama_to_openai(response)