Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

IMPROVEMENT Adds support for new OctoAI endpoints (#13521)

Browse Source 
small fix to add support for new OctoAI LLM endpoints
This commit is contained in:
Bassem Yacoube 2023-11-18 01:15:21 +00:00 committed by GitHub
parent cda1b33270
commit ff382b7b1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/langchain/langchain/llms/octoai_endpoint.py
View File

@ -123,7 +123,7 @@ class OctoAIEndpoint(LLM):
octoai_client = client.Client(token=self.octoai_api_token)
if "model" in _model_kwargs and "llama-2" in _model_kwargs["model"]:
if "model" in _model_kwargs:
parameter_payload = _model_kwargs
parameter_payload["messages"].append(
{"role": "user", "content": prompt}