added support for gpt-5

Author: OKUA1

skllm/llm/anthropic/completion.py

@@ -1,7 +1,7 @@
 from typing import Dict, List, Optional
 from skllm.llm.anthropic.credentials import set_credentials
 from skllm.utils import retry
-from model_constants import ANTHROPIC_CLAUDE_MODEL
+from skllm.model_constants import ANTHROPIC_CLAUDE_MODEL
 
 @retry(max_retries=3)
 def get_chat_completion(

skllm/llm/gpt/clients/openai/completion.py

@@ -5,7 +5,7 @@
     set_credentials,
 )
 from skllm.utils import retry
-from model_constants import OPENAI_GPT_MODEL
+from skllm.model_constants import OPENAI_GPT_MODEL
 
 
 @retry(max_retries=3)
@@ -44,10 +44,11 @@ def get_chat_completion(
         client = set_azure_credentials(key, org)
     else:
         raise ValueError("Invalid API")
-    model_dict = {"model": model}
+    model_dict: dict = {"model": model}
     if json_response and api == "openai":
         model_dict["response_format"] = {"type": "json_object"}
-    completion = client.chat.completions.create(
-        temperature=0.0, messages=messages, **model_dict
-    )
+    if not model.startswith(("gpt-o", "gpt-5")):
+        model_dict["temperature"] = 0.0
+    print("Setting the temperature ", model_dict.get("temperature"))
+    completion = client.chat.completions.create(messages=messages, **model_dict)  # type: ignore
     return completion

skllm/llm/gpt/clients/openai/credentials.py

@@ -5,7 +5,7 @@
 from skllm.config import SKLLMConfig as _Config
 
 
-def set_credentials(key: str, org: str) -> None:
+def set_credentials(key: str, org: str) -> OpenAI:
     """Set the OpenAI key and organization.
 
     Parameters
@@ -20,7 +20,7 @@ def set_credentials(key: str, org: str) -> None:
     return client
 
 
-def set_azure_credentials(key: str, org: str) -> None:
+def set_azure_credentials(key: str, org: str) -> AzureOpenAI:
     """Sets OpenAI credentials for Azure.
 
     Parameters

skllm/llm/gpt/clients/openai/embedding.py

@@ -2,7 +2,7 @@
 from skllm.utils import retry
 import openai
 from openai import OpenAI
-from model_constants import OPENAI_EMBEDDING_MODEL
+from skllm.model_constants import OPENAI_EMBEDDING_MODEL
 
 
 @retry(max_retries=3)

skllm/llm/gpt/completion.py

@@ -7,13 +7,13 @@
 )
 from skllm.llm.gpt.utils import split_to_api_and_model
 from skllm.config import SKLLMConfig as _Config
-from model_constants import OPENAI_GPT_MODEL
+from skllm.model_constants import OPENAI_GPT_MODEL
 
 
 def get_chat_completion(
     messages: dict,
-    openai_key: str = None,
-    openai_org: str = None,
+    openai_key: str,
+    openai_org: str,
     model: str = OPENAI_GPT_MODEL,
     json_response: bool = False,
 ):

skllm/llm/gpt/embedding.py

@@ -1,6 +1,6 @@
 from skllm.llm.gpt.clients.openai.embedding import get_embedding as _oai_get_embedding
 from skllm.llm.gpt.utils import split_to_api_and_model
-from model_constants import OPENAI_EMBEDDING_MODEL
+from skllm.model_constants import OPENAI_EMBEDDING_MODEL
 
 
 def get_embedding(

skllm/llm/gpt/mixin.py

@@ -21,7 +21,7 @@
 import numpy as np
 from tqdm import tqdm
 import json
-from model_constants import OPENAI_GPT_TUNABLE_MODEL
+from skllm.model_constants import OPENAI_GPT_TUNABLE_MODEL
 
 
 def construct_message(role: str, content: str) -> dict:

skllm/model_constants.py

@@ -1,10 +1,10 @@
 # OpenAI models
-OPENAI_GPT_MODEL = "gpt-3.5-turbo"
+OPENAI_GPT_MODEL = "gpt-4.1"
 OPENAI_GPT_TUNABLE_MODEL = "gpt-3.5-turbo-0613"
 OPENAI_EMBEDDING_MODEL = "text-embedding-ada-002"
 
 # Anthropic (Claude) models
 ANTHROPIC_CLAUDE_MODEL = "claude-3-haiku-20240307"
 
 # Vertex AI models
-VERTEX_DEFAULT_MODEL = "text-bison@002"      
+VERTEX_DEFAULT_MODEL = "text-bison@002"

skllm/models/anthropic/classification/few_shot.py

@@ -9,7 +9,7 @@
 from skllm.models._base.vectorizer import BaseVectorizer
 from skllm.memory.base import IndexConstructor
 from typing import Optional
-from model_constants import ANTHROPIC_CLAUDE_MODEL, OPENAI_EMBEDDING_MODEL
+from skllm.model_constants import ANTHROPIC_CLAUDE_MODEL, OPENAI_EMBEDDING_MODEL
 
 
 class FewShotClaudeClassifier(BaseFewShotClassifier, ClaudeClassifierMixin, SingleLabelMixin):

skllm/models/anthropic/classification/zero_shot.py

@@ -6,7 +6,7 @@
 )
 from skllm.llm.anthropic.mixin import ClaudeClassifierMixin as _ClaudeClassifierMixin
 from typing import Optional
-from model_constants import ANTHROPIC_CLAUDE_MODEL
+from skllm.model_constants import ANTHROPIC_CLAUDE_MODEL
 
 
 class ZeroShotClaudeClassifier(