Update huggingface_inference_node.py

huggingface_inference_node.py

@@ -4,11 +4,9 @@ from datetime import datetime
 
 import anthropic
 from groq import Groq
-from openai import OpenAI
 
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 groq_api_key = os.getenv("GROQ_API_KEY")
-sambanova_api_key = os.getenv("SAMBANOVA_API_KEY")
 
 
 class LLMInferenceNode:
@@ -18,9 +16,6 @@ class LLMInferenceNode:
             api_key=huggingface_token,
         )
         self.groq_client = Groq(api_key=groq_api_key)
-        self.sambanova_client = OpenAI(
-            api_key=sambanova_api_key,
-            base_url="https://api.sambanova.ai/v1",
         )
 
     def generate(
@@ -108,35 +103,6 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
                 )
                 output = response.choices[0].message.content.strip()
 
-            elif provider == "OpenAI":
-                openai_client = OpenAI(api_key=api_key)
-                response = openai_client.chat.completions.create(
-                    model=model or "gpt-4",
-                    max_tokens=1024,
-                    temperature=0.7,
-                    messages=[
-                        {"role": "system", "content": system_message},
-                        {"role": "user", "content": user_message},
-                    ],
-                )
-                output = response.choices[0].message.content.strip()
-
-            elif provider == "Anthropic":
-                anthropic_client = anthropic.Anthropic(api_key=api_key)
-                response = anthropic_client.messages.create(
-                    model=model or "claude-3-5-sonnet-20240620",
-                    max_tokens=1024,
-                    temperature=0.7,
-                    system=system_message,
-                    messages=[
-                        {
-                            "role": "user",
-                            "content": [{"type": "text", "text": user_message}],
-                        }
-                    ],
-                )
-                output = response.content[0].text
-
             elif provider == "Groq":
                 response = self.groq_client.chat.completions.create(
                     model=model or "llama-3.1-70b-versatile",
@@ -148,15 +114,6 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
                     ],
                 )
                 output = response.choices[0].message.content.strip()
-
-            elif provider == "SambaNova":
-                response = self.sambanova_client.chat.completions.create(
-                    model=model or "Meta-Llama-3.1-70B-Instruct",
-                    max_tokens=1024,
-                    temperature=0.7,
-                    messages=[
-                        {"role": "system", "content": system_message},
-                        {"role": "user", "content": user_message},
                     ],
                 )
                 output = response.choices[0].message.content.strip()