Add default value to the logprobs parameter

.gitignore

@@ -8,7 +8,7 @@ node_modules/
 *.log
 
 # Python virtualenv
-.venv
+.venv*
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

autogen/oai/client.py

@@ -6,6 +6,7 @@
 import logging
 import inspect
 from flaml.automl.logger import logger_formatter
+from pydantic import ValidationError
 
 from autogen.oai.openai_utils import get_key, oai_price1k
 from autogen.token_count_utils import count_token
@@ -329,15 +330,27 @@ def _completions_create(self, client, params):
                 ),
             )
             for i in range(len(response_contents)):
-                response.choices.append(
-                    Choice(
+                try:
+                    # OpenAI versions 0.1.5 and above
+                    choice = Choice(
                         index=i,
                         finish_reason=finish_reasons[i],
                         message=ChatCompletionMessage(
                             role="assistant", content=response_contents[i], function_call=None
                         ),
+                        logprobs=None,
                     )
-                )
+                except ValidationError:
+                    # OpenAI version up to 0.1.4
+                    choice = Choice(
+                        index=i,
+                        finish_reason=finish_reasons[i],
+                        message=ChatCompletionMessage(
+                            role="assistant", content=response_contents[i], function_call=None
+                        ),
+                    )
+
+                response.choices.append(choice)
         else:
             # If streaming is not enabled or using functions, send a regular chat completion request
             # Functions are not supported, so ensure streaming is disabled

test/oai/test_client.py

@@ -21,7 +21,7 @@ def test_aoai_chat_completion():
     config_list = config_list_from_json(
         env_or_file=OAI_CONFIG_LIST,
         file_location=KEY_LOC,
-        filter_dict={"api_type": ["azure"], "model": ["gpt-3.5-turbo"]},
+        filter_dict={"api_type": ["azure"], "model": ["gpt-3.5-turbo", "gpt-35-turbo"]},
     )
     client = OpenAIWrapper(config_list=config_list)
     # for config in config_list:
@@ -38,7 +38,7 @@ def test_oai_tool_calling_extraction():
     config_list = config_list_from_json(
         env_or_file=OAI_CONFIG_LIST,
         file_location=KEY_LOC,
-        filter_dict={"api_type": ["azure"], "model": ["gpt-3.5-turbo"]},
+        filter_dict={"api_type": ["azure"], "model": ["gpt-3.5-turbo", "gpt-35-turbo"]},
     )
     client = OpenAIWrapper(config_list=config_list)
     response = client.create(