From 5b76f2cde01e063dfdf6ed645cae5341ffb78026 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Mon, 16 Dec 2024 19:07:41 +0100
Subject: Fix docker build and fix api_base issue in OpenaiAPI providers

---
 g4f/Provider/needs_auth/DeepInfra.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'g4f/Provider/needs_auth/DeepInfra.py')

diff --git a/g4f/Provider/needs_auth/DeepInfra.py b/g4f/Provider/needs_auth/DeepInfra.py
index 035effb0..3b5b6227 100644
--- a/g4f/Provider/needs_auth/DeepInfra.py
+++ b/g4f/Provider/needs_auth/DeepInfra.py
@@ -8,6 +8,7 @@ class DeepInfra(OpenaiAPI):
     label = "DeepInfra"
     url = "https://deepinfra.com"
     working = True
+    api_base = "https://api.deepinfra.com/v1/openai",
     needs_auth = True
     supports_stream = True
     supports_message_history = True
@@ -27,7 +28,6 @@ class DeepInfra(OpenaiAPI):
         model: str,
         messages: Messages,
         stream: bool,
-        api_base: str = "https://api.deepinfra.com/v1/openai",
         temperature: float = 0.7,
         max_tokens: int = 1028,
         **kwargs
@@ -50,7 +50,6 @@ class DeepInfra(OpenaiAPI):
         return super().create_async_generator(
             model, messages,
             stream=stream,
-            api_base=api_base,
             temperature=temperature,
             max_tokens=max_tokens,
             headers=headers,
-- 
cgit v1.2.3