langfun 0.1.1.dev20240812__py3-none-any.whl → 0.1.1.dev20240817__py3-none-any.whl

langfun/core/eval/__init__.py

@@ -18,7 +18,7 @@
 
 from langfun.core.eval.base import register
 from langfun.core.eval.base import registered_names
-from langfun.core.eval.base import get_evaluation
+from langfun.core.eval.base import get_evaluations
 from langfun.core.eval.base import get
 from langfun.core.eval.base import run
 

langfun/core/eval/base.py

@@ -2159,14 +2159,17 @@ class _NamedEvaluationRegistry:
     """Returns all registered names."""
     return sorted(self._registry.keys())
 
-  def get(self, name: str) -> Type[Evaluable]:
+  def get(self, name: str) -> list[Type[Evaluable]]:
     """Gets an evaluation by name."""
-    if name not in self._registry:
-      raise ValueError(
-          f'Evaluation {name!r} not found. '
-          'Did you forget to import the module that registers it?'
-      )
-    return self._registry[name]
+    matches = []
+    if name in self._registry:
+      matches.append(self._registry[name])
+    else:
+      regex = re.compile(name)
+      for key, cls in self._registry.items():
+        if regex.match(key):
+          matches.append(cls)
+    return matches
 
   def register(
       self,
@@ -2185,11 +2188,11 @@ def registered_names() -> list[str]:
   return _eval_registry.names()
 
 
-def get_evaluation(evaluation: str | Evaluable) -> Evaluable:
+def get_evaluations(evaluation: str | Evaluable) -> list[Evaluable]:
   """Gets an evaluation experiment by name."""
   if isinstance(evaluation, str):
-    return _eval_registry.get(evaluation)()
-  return evaluation
+    return [e() for e in _eval_registry.get(evaluation)]
+  return [evaluation]
 
 
 def register(name: str):
@@ -2257,8 +2260,14 @@ def get(
   Returns:
     A suite of selected `lf.eval.Evaluation` objects.
   """
-  evaluations = [get_evaluation(e) for e in evaluations]
-  suite = Suite(evaluations, root_dir=root_dir)
+  matches = []
+  for e in evaluations:
+    matches.extend(get_evaluations(e))
+
+  if not matches:
+    raise ValueError('No evaluations found.')
+
+  suite = Suite(matches, root_dir=root_dir)
   if patches:
     suite = pg.patch(suite, patches)
 

langfun/core/eval/base_test.py

@@ -765,7 +765,7 @@ class NamedEvaluationTest(unittest.TestCase):
       ])
       schema_fn = answer_schema()
 
-    evaluation = base.get_evaluation('named_eval/class_test')
+    [evaluation] = base.get_evaluations('named_eval/class_test')
     self.assertIsInstance(evaluation, MyEval)
     self.assertIsNone(evaluation.dir)
     self.assertIsNone(evaluation.root_dir)
@@ -793,13 +793,15 @@ class NamedEvaluationTest(unittest.TestCase):
       )
 
     self.assertTrue(issubclass(my_eval, base.Evaluable))
-    evaluation = base.get_evaluation('named_eval/functor_test')
+    [evaluation] = base.get_evaluations('named_eval/functor_test')
     self.assertIn('named_eval/functor_test', base.registered_names())
     self.assertIsInstance(evaluation, my_eval)
     self.assertIsNone(evaluation.root_dir, None)
 
-    with self.assertRaisesRegex(ValueError, 'Evaluation .* not found'):
-      base.get_evaluation('named_eval/non_existent')
+    self.assertTrue(
+        pg.eq(base.get_evaluations('named_eval/functor.*'), [evaluation])
+    )
+    self.assertEqual(base.get_evaluations('named_eval/non_existent'), [])
 
     with self.assertRaisesRegex(TypeError, 'The return value .*'):
       @base.register('named_eval/bad_return_type')
@@ -841,6 +843,9 @@ class NamedEvaluationTest(unittest.TestCase):
     )
     self.assertTrue(pg.eq(e.leaf_nodes[0].lm, fake.StaticResponse('efg')))
 
+    with self.assertRaisesRegex(ValueError, 'No evaluations found'):
+      base.run(tempfile.gettempdir(), ['test/non_existent'])
+
 
 if __name__ == '__main__':
   unittest.main()

langfun/core/llms/__init__.py

@@ -42,6 +42,7 @@ from langfun.core.llms.openai import OpenAI
 from langfun.core.llms.openai import Gpt4oMini
 from langfun.core.llms.openai import Gpt4oMini_20240718
 from langfun.core.llms.openai import Gpt4o
+from langfun.core.llms.openai import Gpt4o_20240806
 from langfun.core.llms.openai import Gpt4o_20240513
 
 from langfun.core.llms.openai import Gpt4Turbo

langfun/core/llms/openai.py

@@ -37,6 +37,7 @@ SUPPORTED_MODELS_AND_SETTINGS = {
     'gpt-4o-mini': pg.Dict(rpm=10000, tpm=5000000),
     'gpt-4o-mini-2024-07-18': pg.Dict(rpm=10000, tpm=5000000),
     'gpt-4o': pg.Dict(rpm=10000, tpm=5000000),
+    'gpt-4o-2024-08-06': pg.Dict(rpm=10000, tpm=5000000),
     'gpt-4o-2024-05-13': pg.Dict(rpm=10000, tpm=5000000),
     # GPT-4-Turbo models
     'gpt-4-turbo': pg.Dict(rpm=10000, tpm=2000000),
@@ -178,6 +179,8 @@ class OpenAI(lf.LanguageModel):
       args['top_p'] = options.top_p
     if options.stop:
       args['stop'] = options.stop
+    if options.random_seed is not None:
+      args['seed'] = options.random_seed
     return args
 
   def _sample(self, prompts: list[lf.Message]) -> list[lf.LMSamplingResult]:
@@ -231,10 +234,10 @@ class OpenAI(lf.LanguageModel):
   def _chat_complete_batch(
       self, prompts: list[lf.Message]
   ) -> list[lf.LMSamplingResult]:
-    def _open_ai_chat_completion(prompt: lf.Message):
+    def _content_from_message(message: lf.Message):
       if self.multimodal:
         content = []
-        for chunk in prompt.chunk():
+        for chunk in message.chunk():
           if isinstance(chunk, str):
             item = dict(type='text', text=chunk)
           elif isinstance(chunk, lf_modalities.Image):
@@ -244,14 +247,56 @@ class OpenAI(lf.LanguageModel):
             raise ValueError(f'Unsupported modality object: {chunk!r}.')
           content.append(item)
       else:
-        content = prompt.text
+        content = message.text
+      return content
 
-      response = openai.ChatCompletion.create(
-          # TODO(daiyip): support conversation history and system prompt.
-          messages=[{'role': 'user', 'content': content}],
-          **self._get_request_args(self.sampling_options),
+    def _open_ai_chat_completion(prompt: lf.Message):
+      request_args = self._get_request_args(self.sampling_options)
+      # Users could use `metadata_json_schema` to pass additional
+      # request arguments.
+      json_schema = prompt.metadata.get('json_schema')
+      if json_schema is not None:
+        if not isinstance(json_schema, dict):
+          raise ValueError(
+              f'`json_schema` must be a dict, got {json_schema!r}.'
+          )
+        if 'title' not in json_schema:
+          raise ValueError(
+              f'The root of `json_schema` must have a `title` field, '
+              f'got {json_schema!r}.'
+          )
+        request_args.update(
+            response_format=dict(
+                type='json_schema',
+                json_schema=dict(
+                    schema=json_schema,
+                    name=json_schema['title'],
+                    strict=True,
+                )
+            )
+        )
+        prompt.metadata.formatted_text = (
+            prompt.text
+            + '\n\n [RESPONSE FORMAT (not part of prompt)]\n'
+            + pg.to_json_str(request_args['response_format'], json_indent=2)
+        )
+
+      # Prepare messages.
+      messages = []
+      # Users could use `metadata_system_message` to pass system message.
+      system_message = prompt.metadata.get('system_message')
+      if system_message:
+        system_message = lf.SystemMessage.from_value(system_message)
+        messages.append(
+            dict(role='system', content=_content_from_message(system_message))
+        )
+      messages.append(dict(role='user', content=_content_from_message(prompt)))
+
+      response = cast(
+          openai_object.OpenAIObject,
+          openai.ChatCompletion.create(messages=messages, **request_args)
       )
-      response = cast(openai_object.OpenAIObject, response)
+
       samples = []
       for choice in response.choices:
         logprobs = None
@@ -367,8 +412,14 @@ class Gpt4o(OpenAI):
   multimodal = True
 
 
+class Gpt4o_20240806(OpenAI):     # pylint:disable=invalid-name
+  """GPT-4o version 2024-08-06."""
+  model = 'gpt-4o-2024-08-06'
+  multimodal = True
+
+
 class Gpt4o_20240513(OpenAI):     # pylint:disable=invalid-name
-  """GPT-4o."""
+  """GPT-4o version 2024-05-13."""
   model = 'gpt-4o-2024-05-13'
   multimodal = True
 

langfun/core/llms/openai_test.py

@@ -43,12 +43,23 @@ def mock_completion_query(prompt, *, n=1, **kwargs):
 
 
 def mock_chat_completion_query(messages, *, n=1, **kwargs):
-  del messages, kwargs
+  if len(messages) > 1:
+    system_message = f' system={messages[0]["content"]}'
+  else:
+    system_message = ''
+
+  if 'response_format' in kwargs:
+    response_format = f' format={kwargs["response_format"]["type"]}'
+  else:
+    response_format = ''
+
   choices = []
   for k in range(n):
     choices.append(pg.Dict(
         message=pg.Dict(
-            content=f'Sample {k} for message.'
+            content=(
+                f'Sample {k} for message.{system_message}{response_format}'
+            )
         ),
         logprobs=None,
     ))
@@ -123,7 +134,9 @@ class OpenAITest(unittest.TestCase):
     )
     self.assertEqual(
         openai.Gpt4(api_key='test_key')._get_request_args(
-            lf.LMSamplingOptions(temperature=1.0, stop=['\n'], n=1)
+            lf.LMSamplingOptions(
+                temperature=1.0, stop=['\n'], n=1, random_seed=123
+            )
         ),
         dict(
             model='gpt-4',
@@ -134,6 +147,7 @@ class OpenAITest(unittest.TestCase):
             stream=False,
             timeout=120.0,
             stop=['\n'],
+            seed=123,
         ),
     )
 
@@ -461,6 +475,52 @@ class OpenAITest(unittest.TestCase):
         ),
     )
 
+  def test_call_with_system_message(self):
+    with mock.patch('openai.ChatCompletion.create') as mock_chat_completion:
+      mock_chat_completion.side_effect = mock_chat_completion_query
+      lm = openai.OpenAI(api_key='test_key', model='gpt-4')
+      self.assertEqual(
+          lm(
+              lf.UserMessage(
+                  'hello',
+                  system_message='hi',
+              ),
+              sampling_options=lf.LMSamplingOptions(n=2)
+          ),
+          'Sample 0 for message. system=hi',
+      )
+
+  def test_call_with_json_schema(self):
+    with mock.patch('openai.ChatCompletion.create') as mock_chat_completion:
+      mock_chat_completion.side_effect = mock_chat_completion_query
+      lm = openai.OpenAI(api_key='test_key', model='gpt-4')
+      self.assertEqual(
+          lm(
+              lf.UserMessage(
+                  'hello',
+                  json_schema={
+                      'type': 'object',
+                      'properties': {
+                          'name': {'type': 'string'},
+                      },
+                      'required': ['name'],
+                      'title': 'Person',
+                  }
+              ),
+              sampling_options=lf.LMSamplingOptions(n=2)
+          ),
+          'Sample 0 for message. format=json_schema',
+      )
+
+    # Test bad json schema.
+    with self.assertRaisesRegex(ValueError, '`json_schema` must be a dict'):
+      lm(lf.UserMessage('hello', json_schema='foo'))
+
+    with self.assertRaisesRegex(
+        ValueError, 'The root of `json_schema` must have a `title` field'
+    ):
+      lm(lf.UserMessage('hello', json_schema={}))
+
 
 if __name__ == '__main__':
   unittest.main()

langfun/core/llms/vertexai.py

@@ -18,18 +18,17 @@ import os
 from typing import Annotated, Any
 
 from google.auth import credentials as credentials_lib
-from google.cloud.aiplatform import aiplatform
 import langfun.core as lf
 from langfun.core import modalities as lf_modalities
 import pyglove as pg
 
 
 SUPPORTED_MODELS_AND_SETTINGS = {
-    'gemini-1.5-pro-001': pg.Dict(api='gemini', rpm=5),
-    'gemini-1.5-flash-001': pg.Dict(api='gemini', rpm=5),
-    'gemini-1.5-pro-preview-0514': pg.Dict(api='gemini', rpm=5),
-    'gemini-1.5-pro-preview-0409': pg.Dict(api='gemini', rpm=5),
-    'gemini-1.5-flash-preview-0514': pg.Dict(api='gemini', rpm=5),
+    'gemini-1.5-pro-001': pg.Dict(api='gemini', rpm=50),
+    'gemini-1.5-flash-001': pg.Dict(api='gemini', rpm=200),
+    'gemini-1.5-pro-preview-0514': pg.Dict(api='gemini', rpm=50),
+    'gemini-1.5-pro-preview-0409': pg.Dict(api='gemini', rpm=50),
+    'gemini-1.5-flash-preview-0514': pg.Dict(api='gemini', rpm=200),
     'gemini-1.0-pro': pg.Dict(api='gemini', rpm=300),
     'gemini-1.0-pro-vision': pg.Dict(api='gemini', rpm=100),
     # PaLM APIs.
@@ -136,16 +135,34 @@ class VertexAI(lf.LanguageModel):
     )
 
   def _generation_config(
-      self, options: lf.LMSamplingOptions
+      self, prompt: lf.Message, options: lf.LMSamplingOptions
   ) -> Any:  # generative_models.GenerationConfig
     """Creates generation config from langfun sampling options."""
     from vertexai import generative_models
+    # Users could use `metadata_json_schema` to pass additional
+    # request arguments.
+    json_schema = prompt.metadata.get('json_schema')
+    response_mime_type = None
+    if json_schema is not None:
+      if not isinstance(json_schema, dict):
+        raise ValueError(
+            f'`json_schema` must be a dict, got {json_schema!r}.'
+        )
+      response_mime_type = 'application/json'
+      prompt.metadata.formatted_text = (
+          prompt.text
+          + '\n\n [RESPONSE FORMAT (not part of prompt)]\n'
+          + pg.to_json_str(json_schema, json_indent=2)
+      )
+
     return generative_models.GenerationConfig(
         temperature=options.temperature,
         top_p=options.top_p,
         top_k=options.top_k,
         max_output_tokens=options.max_tokens,
         stop_sequences=options.stop,
+        response_mime_type=response_mime_type,
+        response_schema=json_schema,
     )
 
   def _content_from_message(
@@ -239,7 +256,9 @@ class VertexAI(lf.LanguageModel):
     input_content = self._content_from_message(prompt)
     response = model.generate_content(
         input_content,
-        generation_config=self._generation_config(self.sampling_options),
+        generation_config=self._generation_config(
+            prompt, self.sampling_options
+        ),
     )
     usage_metadata = response.usage_metadata
     usage = lf.LMSamplingUsage(
@@ -277,7 +296,8 @@ class VertexAI(lf.LanguageModel):
 
   def _sample_endpoint_model(self, prompt: lf.Message) -> lf.LMSamplingResult:
     """Samples a text generation model."""
-    model = aiplatform.Endpoint(self.endpoint_name)
+    from google.cloud.aiplatform import models
+    model = models.Endpoint(self.endpoint_name)
     # TODO(chengrun): Add support for stop_sequences.
     predict_options = dict(
         temperature=self.sampling_options.temperature

langfun/core/llms/vertexai_test.py

@@ -17,7 +17,7 @@ import os
 import unittest
 from unittest import mock
 
-from google.cloud.aiplatform.aiplatform import models as aiplatform_models
+from google.cloud.aiplatform import models as aiplatform_models
 from vertexai import generative_models
 import langfun.core as lf
 from langfun.core import modalities as lf_modalities
@@ -175,6 +175,53 @@ class VertexAITest(unittest.TestCase):
     del os.environ['VERTEXAI_PROJECT']
     del os.environ['VERTEXAI_LOCATION']
 
+  def test_generation_config(self):
+    model = vertexai.VertexAIGeminiPro1()
+    json_schema = {
+        'type': 'object',
+        'properties': {
+            'name': {'type': 'string'},
+        },
+        'required': ['name'],
+        'title': 'Person',
+    }
+    config = model._generation_config(
+        lf.UserMessage('hi', json_schema=json_schema),
+        lf.LMSamplingOptions(
+            temperature=2.0,
+            top_p=1.0,
+            top_k=20,
+            max_tokens=1024,
+            stop=['\n'],
+        ),
+    )
+    self.assertEqual(
+        config.to_dict(),
+        dict(
+            temperature=2.0,
+            top_p=1.0,
+            top_k=20.0,
+            max_output_tokens=1024,
+            stop_sequences=['\n'],
+            response_mime_type='application/json',
+            response_schema={
+                'type_': 'OBJECT',
+                'properties': {
+                    'name': {'type_': 'STRING'}
+                },
+                'required': ['name'],
+                'title': 'Person',
+            }
+        ),
+    )
+    with self.assertRaisesRegex(
+        ValueError, '`json_schema` must be a dict, got'
+    ):
+      model._generation_config(
+          lf.UserMessage('hi', json_schema='not a dict'),
+          lf.LMSamplingOptions(),
+      )
+
   def test_call_generative_model(self):
     with mock.patch(
         'vertexai.generative_models.'
@@ -244,27 +291,31 @@ class VertexAITest(unittest.TestCase):
 
   def test_call_endpoint_model(self):
     with mock.patch(
-        'google.cloud.aiplatform.aiplatform.Endpoint.predict'
-    ) as mock_model_predict:
-
-      mock_model_predict.side_effect = mock_endpoint_predict
-      lm = vertexai.VertexAI(
-          'custom',
-          endpoint_name='123',
-          project='abc',
-          location='us-central1',
-      )
-      self.assertEqual(
-          lm(
-              'hello',
-              temperature=2.0,
-              top_p=1.0,
-              top_k=20,
-              max_tokens=50,
-          ),
-          'This is a response to hello with temperature=2.0, top_p=1.0,'
-          ' top_k=20, max_tokens=50.',
-      )
+        'google.cloud.aiplatform.models.Endpoint.__init__'
+    ) as mock_model_init:
+      mock_model_init.side_effect = lambda *args, **kwargs: None
+      with mock.patch(
+          'google.cloud.aiplatform.models.Endpoint.predict'
+      ) as mock_model_predict:
+
+        mock_model_predict.side_effect = mock_endpoint_predict
+        lm = vertexai.VertexAI(
+            'custom',
+            endpoint_name='123',
+            project='abc',
+            location='us-central1',
+        )
+        self.assertEqual(
+            lm(
+                'hello',
+                temperature=2.0,
+                top_p=1.0,
+                top_k=20,
+                max_tokens=50,
+            ),
+            'This is a response to hello with temperature=2.0, top_p=1.0,'
+            ' top_k=20, max_tokens=50.',
+        )
 
 
 if __name__ == '__main__':

{langfun-0.1.1.dev20240812.dist-info → langfun-0.1.1.dev20240817.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langfun
-Version: 0.1.1.dev20240812
+Version: 0.1.1.dev20240817
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.1.dev20240812.dist-info → langfun-0.1.1.dev20240817.dist-info}/RECORD

@@ -43,16 +43,16 @@ langfun/core/coding/python/parsing.py,sha256=LMg8REP4VDY0YQjtPAGNAW4rKlMNdSXF8m1
 langfun/core/coding/python/parsing_test.py,sha256=9vAWF484kWIm6JZq8NFiMgKUDhXV-deRl1QMmNERfAA,7386
 langfun/core/coding/python/permissions.py,sha256=1QWGHvzL8MM0Ok_auQ9tURqZHtdOfJaDpBzZ29GUE-c,2544
 langfun/core/coding/python/permissions_test.py,sha256=w5EDb8QxpxgJyZkojyzVWQvDfg366zn99-g__6TbPQ0,2699
-langfun/core/eval/__init__.py,sha256=Evt-E4FEhZF2tXL6-byh_AyA7Cc_ZoGmvnN7vkAZedk,1898
-langfun/core/eval/base.py,sha256=D9l1qkg1hS7zxpXKIecmYZf_2CZaDt16TfMl3t7f7Pw,73660
-langfun/core/eval/base_test.py,sha256=cHOTIWVW4Dp8gKKIKcZrAcJ-w84j2GIozTzJoiAX7p4,26743
+langfun/core/eval/__init__.py,sha256=Ogdr9OtTywhhLPHi3AZzOD2mXX2oyaHWflrSTMm96uA,1899
+langfun/core/eval/base.py,sha256=0_iaKuQhS49PlbWqCQ5EABUMKavr2R4ltcJZWCVoZZg,73816
+langfun/core/eval/base_test.py,sha256=p1EfqviHMz_ppQY8FU67h5OCgL0tzhLvXzGIsq0sVyI,26930
 langfun/core/eval/matching.py,sha256=9GX8HfO9jKxgNLAivgy5K88Xhoh6Z7Pptq65pe7vht8,9762
 langfun/core/eval/matching_test.py,sha256=f7iVyXH5KGJBWt4Wp14Bt9J3X59A6Ayfog9MbuFvPew,5532
 langfun/core/eval/patching.py,sha256=R0s2eAd1m97exQt06dmUL0V_MBG0W2Hxg7fhNB7cXW0,3866
 langfun/core/eval/patching_test.py,sha256=8kCd54Egjju22FMgtJuxEsrXkW8ifs-UUBHtrCG1L6w,4775
 langfun/core/eval/scoring.py,sha256=AlCwEVrU6nvURDB1aPxA2XBUmOjWxuNJDXJoS4-6VbU,6386
 langfun/core/eval/scoring_test.py,sha256=O8olHbrUEg60gMxwOkWzKBJZpZoUlmVnBANX5Se2SXM,4546
-langfun/core/llms/__init__.py,sha256=ggPvkKq8l-B8rN3ZD6d7r3_d5DYAd5fC7FgC6ZI2Wzo,4766
+langfun/core/llms/__init__.py,sha256=a1AV3XWi2gY4UvmmaPP1GapaQxygA6xzQJvVQRp6EPA,4818
 langfun/core/llms/anthropic.py,sha256=Gon3fOi31RhZFgNd0ijyTnKnUdp9hrWrCoSXyO4UaLw,7316
 langfun/core/llms/anthropic_test.py,sha256=T-swuMkfnlgs8Fpif4rtXs579exGk0TsbLMirXDZCkg,5533
 langfun/core/llms/fake.py,sha256=Dd7-6ka9pFf3fcWZyczamjOqQ91MOI-m7We3Oc9Ffmo,2927
@@ -63,12 +63,12 @@ langfun/core/llms/groq.py,sha256=pqtyOZ_1_OJMOg8xATWT_B_SVbuT9nMRf4VkH9GzW8g,630
 langfun/core/llms/groq_test.py,sha256=GYF_Qtq5S1H1TrKH38t6_lkdroqT7v-joYLDKnmS9e0,5274
 langfun/core/llms/llama_cpp.py,sha256=9tXQntSCDtjTF3bnyJrAPCr4N6wycy5nXYvp9uduygE,2843
 langfun/core/llms/llama_cpp_test.py,sha256=MWO_qaOeKjRniGjcaWPDScd7HPaIJemqUZoslrt4FPs,1806
-langfun/core/llms/openai.py,sha256=jILxfFb3vBuyf1u_2-LVfs_wekPF2RVuNFzNVg25pEA,14004
-langfun/core/llms/openai_test.py,sha256=3muDTnW7UBOSHq694Fi2bofqhe8Pkj0Tl8IShoLCTOM,15525
+langfun/core/llms/openai.py,sha256=lRfR2iim7OdzMCJCf1DXB5YVfSwflvUucMWY3dsMaRA,15798
+langfun/core/llms/openai_test.py,sha256=02KeysRppXcAwA4SBUow8hKolFiEU9_lTSdlVHZletM,17518
 langfun/core/llms/rest.py,sha256=laopuq-zD8V-3Y6eFDngftHEbE66VlUkCD2-rvvRaLU,3388
 langfun/core/llms/rest_test.py,sha256=NZ3Nf0XQVpT9kLP5cBVo_yBHLI7vWTYhWQxYEJVMGs4,3472
-langfun/core/llms/vertexai.py,sha256=-YoEUlG19CWIhJb8S6puPqdX9SoiT5NNAItefwdCfsk,13781
-langfun/core/llms/vertexai_test.py,sha256=N3k4N9_bVjC6_Qtg4WO9jYNv8M9xmv5UdODvIKG2upo,8835
+langfun/core/llms/vertexai.py,sha256=tXAnP357XhcsETTnk6M-hH4xyFi7tk6fsaf3tjzsY6E,14501
+langfun/core/llms/vertexai_test.py,sha256=EPR-mB2hNUpvpf7E8m_k5bh04epdQTVUuYU6hPgZyu8,10321
 langfun/core/llms/cache/__init__.py,sha256=QAo3InUMDM_YpteNnVCSejI4zOsnjSMWKJKzkb3VY64,993
 langfun/core/llms/cache/base.py,sha256=cFfYvOIUae842pncqCAsRvqXCk2AnAsRYVx0mcIoAeY,3338
 langfun/core/llms/cache/in_memory.py,sha256=YfFyJEhLs73cUiB0ZfhMxYpdE8Iuxxw-dvMFwGHTSHw,4742
@@ -117,8 +117,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
 langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
 langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
 langfun/core/templates/selfplay_test.py,sha256=rBW2Qr8yi-aWYwoTwRR-n1peKyMX9QXPZXURjLgoiRs,2264
-langfun-0.1.1.dev20240812.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.1.dev20240812.dist-info/METADATA,sha256=Kk1dXZKetEZkE1Ycs26AdvbUuJY6TIA7Z75LKGqDlq0,5234
-langfun-0.1.1.dev20240812.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-langfun-0.1.1.dev20240812.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.1.dev20240812.dist-info/RECORD,,
+langfun-0.1.1.dev20240817.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.1.dev20240817.dist-info/METADATA,sha256=_j3RVHGW7vU400lTH9B4ncDzlqEmCHOs4GfpwO96quE,5234
+langfun-0.1.1.dev20240817.dist-info/WHEEL,sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4,91
+langfun-0.1.1.dev20240817.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.1.dev20240817.dist-info/RECORD,,

{langfun-0.1.1.dev20240812.dist-info → langfun-0.1.1.dev20240817.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.1.0)
+Generator: setuptools (72.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any