format

Change-Id: Ib052095b489a28ca5a2fc7338c8fefd5ac0adbc5

google/generativeai/types/content_types.py

@@ -370,8 +370,8 @@ def _schema_for_function(
  )
  ]
  schema = dict(name=f.__name__, description=f.__doc__)
- if parameters['properties']:
- schema['parameters'] = parameters
+ if parameters["properties"]:
+ schema["parameters"] = parameters
 
  return schema
 

samples/function_calling.py

@@ -35,7 +35,8 @@ def stop_lights():
  light_controls = [enable_lights, set_light_color, stop_lights]
  instruction = (
  "You are a helpful lighting system bot. You can turn lights on and off, "
- "and you can set the color. Do not perform any other tasks.")
+ "and you can set the color. Do not perform any other tasks."
+ )
  model = genai.GenerativeModel(
  "models/gemini-1.5-pro", tools=light_controls, system_instruction=instruction
  )

tests/test_content.py

@@ -380,14 +380,12 @@ def test_to_tools(self, tools):
 
  def test_empty_function(self):
  def no_args():
- print('hello')
+ print("hello")
 
  fd = content_types.to_function_library(no_args).to_proto()[0]
  fd = type(fd).to_dict(fd, including_default_value_fields=False)
  # parameters are not set.
- self.assertEqual({'function_declarations': [{'name': 'no_args'}]}, fd)
-
-
+ self.assertEqual({"function_declarations": [{"name": "no_args"}]}, fd)
 
  @parameterized.named_parameters(
  ["string", "code_execution"],