langfun 0.1.2.dev202503030804__tar.gz → 0.1.2.dev202503050804__tar.gz

{langfun-0.1.2.dev202503030804 → langfun-0.1.2.dev202503050804}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: langfun
-Version: 0.1.2.dev202503030804
+Version: 0.1.2.dev202503050804
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.2.dev202503030804 → langfun-0.1.2.dev202503050804}/langfun/core/concurrent_test.py

@@ -560,6 +560,7 @@ class ConcurrentMapTest(unittest.TestCase):
               fun, [1, 2, 3], timeout=1.5, max_workers=1, show_progress=True
           )
       ], key=lambda x: x[0])
+      string_io.flush()
     self.assertEqual(   # pylint: disable=g-generic-assert
         output,
         [
@@ -570,7 +571,6 @@ class ConcurrentMapTest(unittest.TestCase):
     )
     output = string_io.getvalue()
     self.assertIn('100%', output)
-    self.assertIn('TimeIt=foo (', output)
 
   def test_concurrent_map_with_showing_progress_and_status_fn(self):
     def fun(x):

{langfun-0.1.2.dev202503030804 → langfun-0.1.2.dev202503050804}/langfun/core/llms/__init__.py

@@ -36,6 +36,7 @@ from langfun.core.llms.anthropic import Anthropic
 # Base models by serving platforms.
 from langfun.core.llms.vertexai import VertexAI
 from langfun.core.llms.groq import Groq
+from langfun.core.llms.azure_openai import AzureOpenAI
 
 # Gemini models.
 from langfun.core.llms.google_genai import GenAI

langfun-0.1.2.dev202503050804/langfun/core/llms/azure_openai.py

@@ -0,0 +1,90 @@
+# Copyright 2025 The Langfun Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""OpenAI models hosted on Azure."""
+import os
+from typing import Annotated, Any
+
+import langfun.core as lf
+from langfun.core.llms import openai
+import pyglove as pg
+
+
+@lf.use_init_args(['model', 'deployment_name'])
+@pg.members([('api_endpoint', pg.typing.Str().freeze(''))])
+class AzureOpenAI(openai.OpenAI):
+  """Azure OpenAI model service.
+
+  This service interacts with the Azure OpenAI API to generate chat completions.
+  It uses the deployment_name and API version to construct the endpoint, and
+  authenticates using an API key provided via parameter or the
+  AZURE_OPENAI_API_KEY environment variable.
+
+  Example:
+      lm = AzureOpenAI(
+          model='gpt-4o',
+          deployment_name='gpt-4o',
+          api_version='2024-08-01-preview',
+          azure_endpoint='https://trackname.openai.azure.com/',
+          api_key='token'
+      )
+      response = lf.query(prompt="what the capital of France", lm=lm)
+      print(response)
+  """
+
+  deployment_name: Annotated[
+      str,
+      'The name of the Azure OpenAI deployment.'
+  ] = 'gpt-4'
+
+  api_version: Annotated[
+      str,
+      'The API version for Azure OpenAI.'
+  ] = '2023-05-15'
+
+  azure_endpoint: Annotated[
+      str,
+      (
+          'The base URL for Azure OpenAI '
+          '(e.g. "https://<your-resource>.openai.azure.com/")'
+      )
+  ] = 'https://api.openai.azure.com/'
+
+  def _initialize(self):
+    # Authentication
+    self._api_key = self.api_key or os.environ.get('AZURE_OPENAI_API_KEY')
+    if not self._api_key:
+      raise ValueError(
+          'Azure OpenAI requires an API key. Please provide '
+          'via `api_key` argument or AZURE_OPENAI_API_KEY '
+          'environment variable.'
+      )
+
+    # Endpoint construction
+    self._api_endpoint = (
+        f"{self.azure_endpoint.rstrip('/')}/openai/deployments/"
+        f"{self.deployment_name}/chat/completions"
+        f"?api-version={self.api_version}"
+    )
+
+  @property
+  def api_endpoint(self) -> str:
+    return self._api_endpoint
+
+  @property
+  def headers(self) -> dict[str, Any]:
+    headers = super().headers
+    headers.update({
+        'api-key': self._api_key,
+    })
+    return headers

langfun-0.1.2.dev202503050804/langfun/core/llms/azure_openai_test.py

@@ -0,0 +1,47 @@
+# Copyright 2025 The Langfun Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import os
+import unittest
+from langfun.core.llms import azure_openai
+
+
+class AzureOpenAITest(unittest.TestCase):
+  def test_api_key_missing(self):
+    # Ensure that ValueError is raised when API key is not provided
+    os.environ.pop('AZURE_OPENAI_API_KEY', None)
+    with self.assertRaisesRegex(ValueError, 'Azure OpenAI requires an API key'):
+      azure = azure_openai.AzureOpenAI(model='gpt-4')
+      azure._initialize()
+
+  def test_api_endpoint(self):
+    # Test that api_endpoint is properly constructed
+    azure = azure_openai.AzureOpenAI(model='gpt-4', api_key='test_key')
+    azure._initialize()
+    expected = (
+        'https://api.openai.azure.com/openai/deployments/gpt-4/chat/'
+        'completions?api-version=2023-05-15'
+    )
+    self.assertEqual(azure.api_endpoint, expected)
+
+  def test_headers(self):
+    # Test that headers contain the api-key with correct value
+    azure = azure_openai.AzureOpenAI(model='gpt-4', api_key='test_key')
+    azure._initialize()
+    headers = azure.headers
+    self.assertIn('api-key', headers)
+    self.assertEqual(headers.get('api-key'), 'test_key')
+
+
+if __name__ == '__main__':
+  unittest.main()

{langfun-0.1.2.dev202503030804 → langfun-0.1.2.dev202503050804}/langfun.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: langfun
-Version: 0.1.2.dev202503030804
+Version: 0.1.2.dev202503050804
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.2.dev202503030804 → langfun-0.1.2.dev202503050804}/langfun.egg-info/SOURCES.txt

@@ -84,6 +84,8 @@ langfun/core/eval/v2/runners_test.py
 langfun/core/llms/__init__.py
 langfun/core/llms/anthropic.py
 langfun/core/llms/anthropic_test.py
+langfun/core/llms/azure_openai.py
+langfun/core/llms/azure_openai_test.py
 langfun/core/llms/compositional.py
 langfun/core/llms/compositional_test.py
 langfun/core/llms/deepseek.py