Update api_handler.py

api_handler.py

@@ -88,6 +88,39 @@ class APIHandler:
             print(f"Błąd pobierania modeli: {e}")
             return ["gpt-4o", "gpt-4", "gpt-3.5-turbo"]
 
+    def _get_model_parameters(self, model, temperature, max_tokens, top_p):
+        """
+        Przygotowuje parametry API dla danego modelu
+
+        Args:
+            model: Nazwa modelu
+            temperature: Temperatura (0.0-2.0)
+            max_tokens: Maksymalna długość odpowiedzi
+            top_p: Nucleus sampling parameter
+
+        Returns:
+            dict: Słownik z parametrami gotowymi do przekazania do API
+        """
+        # Modele o1 wymagają innych parametrów
+        is_o1_model = model.startswith('o1')
+
+        params = {
+            "model": model,
+            "messages": []  # Będzie wypełnione w generate_response
+        }
+
+        if is_o1_model:
+            # Modele o1 używają max_completion_tokens zamiast max_tokens
+            # i nie akceptują temperature/top_p
+            params["max_completion_tokens"] = max_tokens
+        else:
+            # Standardowe modele (gpt-4, gpt-3.5, fine-tuned)
+            params["max_tokens"] = max_tokens
+            params["temperature"] = temperature
+            params["top_p"] = top_p
+
+        return params
+
     def generate_response(self, prompt, model="gpt-4o", temperature=0.1, max_tokens=2000, top_p=1.0):
         """
         Generuje odpowiedź z OpenAI API
@@ -95,9 +128,9 @@ class APIHandler:
         Args:
             prompt: Tekst promptu systemowego
             model: Model OpenAI
-            temperature: Temperatura (0.0-2.0)
+            temperature: Temperatura (0.0-2.0) - ignorowane dla modeli o1
             max_tokens: Maksymalna długość odpowiedzi
-            top_p: Nucleus sampling parameter
+            top_p: Nucleus sampling parameter - ignorowane dla modeli o1
 
         Returns:
             str: Wygenerowana odpowiedź lub komunikat błędu
@@ -106,16 +139,17 @@ class APIHandler:
             return "ERROR: Brak połączenia z API (nieprawidłowy klucz)"
 
         try:
-            response = self.client.chat.completions.create(
-                model=model,
-                messages=[
-                    {"role": "system", "content": prompt},
-                    {"role": "user", "content": "Please provide your response based on the system prompt."}
-                ],
-                temperature=temperature,
-                max_tokens=max_tokens,
-                top_p=top_p
-            )
+            # Przygotuj parametry dostosowane do modelu
+            params = self._get_model_parameters(model, temperature, max_tokens, top_p)
+
+            # Dodaj wiadomości
+            params["messages"] = [
+                {"role": "system", "content": prompt},
+                {"role": "user", "content": "Please provide your response based on the system prompt."}
+            ]
+
+            # Wywołaj API z odpowiednimi parametrami
+            response = self.client.chat.completions.create(**params)
             return response.choices[0].message.content
         except Exception as e:
             error_msg = str(e)
@@ -141,12 +175,15 @@ class APIHandler:
         Returns:
             float: Szacunkowy koszt w USD
         """
-        # Ceny za 1M tokenów (stan na 2025)
+        # Ceny za 1M tokenów (stan na grudzień 2024)
         pricing = {
             "gpt-4o": {"input": 2.50, "output": 10.00},
             "gpt-4-turbo": {"input": 10.00, "output": 30.00},
             "gpt-4": {"input": 30.00, "output": 60.00},
-            "gpt-3.5-turbo": {"input": 0.50, "output": 1.50}
+            "gpt-3.5-turbo": {"input": 0.50, "output": 1.50},
+            "o1-preview": {"input": 15.00, "output": 60.00},
+            "o1-mini": {"input": 3.00, "output": 12.00},
+            "o1": {"input": 15.00, "output": 60.00}
         }
 
         # Znajdź odpowiedni cennik