update to take api key or token

Author: SentienceDEV

sentience/__init__.py

@@ -64,6 +64,7 @@
 from .inspector import Inspector, inspect
 from .llm_provider import (
     AnthropicProvider,
+    DeepInfraProvider,
     LLMProvider,
     LLMResponse,
     LocalLLMProvider,
@@ -227,6 +228,7 @@
     "LLMResponse",
     "OpenAIProvider",
     "AnthropicProvider",
+    "DeepInfraProvider",
     "LocalLLMProvider",
     "LocalVisionLLMProvider",
     "MLXVLMProvider",

sentience/llm_provider.py

@@ -326,7 +326,7 @@ class DeepInfraProvider(OpenAIProvider):
     Uses DeepInfra's OpenAI-compatible endpoint:
     https://api.deepinfra.com/v1/openai
 
-    API token is read from DEEPINFRA_TOKEN if not provided.
+    API token is read from DEEPINFRA_TOKEN or DEEPINFRA_API_KEY if not provided.
     """
 
     def __init__(
@@ -335,7 +335,9 @@ def __init__(
         model: str = "meta-llama/Meta-Llama-3-8B-Instruct",
         base_url: str = "https://api.deepinfra.com/v1/openai",
     ):
-        api_key = get_api_key_from_env(["DEEPINFRA_TOKEN"], api_key)
+        api_key = get_api_key_from_env(
+            ["DEEPINFRA_TOKEN", "DEEPINFRA_API_KEY"], api_key
+        )
         super().__init__(api_key=api_key, model=model, base_url=base_url)