From ff382b7b1b2ff50a912c92c2d2184ba8a1109bec Mon Sep 17 00:00:00 2001
From: Bassem Yacoube <125713079+AI-Bassem@users.noreply.github.com>
Date: Sat, 18 Nov 2023 01:15:21 +0000
Subject: [PATCH] IMPROVEMENT Adds support for new OctoAI endpoints (#13521)

small fix to add support for new OctoAI LLM endpoints
---
 libs/langchain/langchain/llms/octoai_endpoint.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libs/langchain/langchain/llms/octoai_endpoint.py b/libs/langchain/langchain/llms/octoai_endpoint.py
index 055ab8ac754..450b0343edb 100644
--- a/libs/langchain/langchain/llms/octoai_endpoint.py
+++ b/libs/langchain/langchain/llms/octoai_endpoint.py
@@ -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}