Skip to content

Commit

Permalink
IMPROVEMENT Adds support for new OctoAI endpoints (#13521)
Browse files Browse the repository at this point in the history
small fix to add support for new OctoAI LLM endpoints
  • Loading branch information
AI-Bassem authored Nov 18, 2023
1 parent cda1b33 commit ff382b7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion libs/langchain/langchain/llms/octoai_endpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def _call(

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}
Expand Down

0 comments on commit ff382b7

Please sign in to comment.