Microsoft LLM feature

Author: Daggx

edenai_apis/apis/microsoft/info.json

@@ -1761,6 +1761,11 @@
       "version": "Azure AI Foundry"
     }
   },
+  "llm": {
+    "chat": {
+      "version": "Azure AI Foundry"
+    }
+  },
   "_metadata": {
     "privacy_url": "https://privacy.microsoft.com/en-us"
   }

edenai_apis/apis/microsoft/microsoft_api.py

@@ -10,6 +10,7 @@
 from edenai_apis.apis.microsoft.microsoft_text_api import MicrosoftTextApi
 from edenai_apis.apis.microsoft.microsoft_translation_api import MicrosoftTranslationApi
 from edenai_apis.apis.microsoft.microsoft_multimodal_api import MicrosoftMultimodalApi
+from edenai_apis.apis.microsoft.microsoft_llm_api import MicrosoftLLMApi
 from edenai_apis.features.provider.provider_interface import ProviderInterface
 from edenai_apis.loaders.data_loader import ProviderDataEnum
 from edenai_apis.loaders.loaders import load_provider
@@ -24,6 +25,7 @@ class MicrosoftApi(
     MicrosoftOcrApi,
     MicrosoftAudioApi,
     MicrosoftMultimodalApi,
+    MicrosoftLLMApi,
 ):
     provider_name = "microsoft"
 

edenai_apis/apis/microsoft/microsoft_llm_api.py

@@ -0,0 +1,85 @@
+from typing import List, Type, Union, Optional
+
+import httpx
+from openai import BaseModel
+from edenai_apis.features.llm.llm_interface import LlmInterface
+from edenai_apis.features.llm.chat.chat_dataclass import ChatDataClass
+
+
+class MicrosoftLLMApi(LlmInterface):
+    def llm__chat(
+        self,
+        messages: List = [],
+        model: Optional[str] = None,
+        # Optional OpenAI params: see https://platform.openai.com/docs/api-reference/chat/create
+        timeout: Optional[Union[float, str, httpx.Timeout]] = None,
+        temperature: Optional[float] = None,
+        top_p: Optional[float] = None,
+        n: Optional[int] = None,
+        stream: Optional[bool] = None,
+        stream_options: Optional[dict] = None,
+        stop: Optional[str] = None,
+        stop_sequences: Optional[any] = None,
+        max_tokens: Optional[int] = None,
+        presence_penalty: Optional[float] = None,
+        frequency_penalty: Optional[float] = None,
+        logit_bias: Optional[dict] = None,
+        # openai v1.0+ new params
+        response_format: Optional[
+            Union[dict, Type[BaseModel]]
+        ] = None,  # Structured outputs
+        seed: Optional[int] = None,
+        tools: Optional[List] = None,
+        tool_choice: Optional[Union[str, dict]] = None,
+        logprobs: Optional[bool] = None,
+        top_logprobs: Optional[int] = None,
+        parallel_tool_calls: Optional[bool] = None,
+        deployment_id=None,
+        extra_headers: Optional[dict] = None,
+        # soon to be deprecated params by OpenAI -> This should be replaced by tools
+        functions: Optional[List] = None,
+        function_call: Optional[str] = None,
+        base_url: Optional[str] = None,
+        api_version: Optional[str] = None,
+        api_key: Optional[str] = None,
+        model_list: Optional[list] = None,  # pass in a list of api_base,keys, etc.
+        drop_invalid_params: bool = True,  # If true, all the invalid parameters will be ignored (dropped) before sending to the model
+        user: str | None = None,
+        # Optional parameters
+        **kwargs,
+    ) -> ChatDataClass:
+        response = self.llm_client.completion(
+            messages=messages,
+            model=model,
+            timeout=timeout,
+            temperature=temperature,
+            top_p=top_p,
+            n=n,
+            stream=stream,
+            stream_options=stream_options,
+            stop=stop,
+            stop_sequences=stop_sequences,
+            max_tokens=max_tokens,
+            presence_penalty=presence_penalty,
+            frequency_penalty=frequency_penalty,
+            logit_bias=logit_bias,
+            response_format=response_format,
+            seed=seed,
+            tools=tools,
+            tool_choice=tool_choice,
+            logprobs=logprobs,
+            top_logprobs=top_logprobs,
+            parallel_tool_calls=parallel_tool_calls,
+            deployment_id=deployment_id,
+            extra_headers=extra_headers,
+            functions=functions,
+            function_call=function_call,
+            base_url=base_url,
+            api_version=api_version,
+            api_key=api_key,
+            model_list=model_list,
+            drop_invalid_params=drop_invalid_params,
+            user=user,
+            **kwargs,
+        )
+        return response