From 9b7465a222ee3c1873382144bcc15000109f6ff6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebasti=C3=A1n=20Est=C3=A9vez?= Date: Wed, 15 May 2024 17:06:54 -0400 Subject: [PATCH] Another dictionary changed size during iteration error ``` ImportError while loading conftest '/astra-assistants-api/tests/openai-sdk/conftest.py'. conftest.py:13: in from impl.astra_vector import CassandraClient ../../impl/astra_vector.py:45: in from impl.services.inference_utils import get_embeddings ../../impl/services/inference_utils.py:5: in import litellm .cache/pypoetry/virtualenvs/astra-assistants-api-eiSmbCzm-py3.10/lib/python3.10/site-packages/litellm/__init__.py:678: in from .main import * # type: ignore .cache/pypoetry/virtualenvs/astra-assistants-api-eiSmbCzm-py3.10/lib/python3.10/site-packages/litellm/main.py:73: in from .llms.azure_text import AzureTextCompletion .cache/pypoetry/virtualenvs/astra-assistants-api-eiSmbCzm-py3.10/lib/python3.10/site-packages/litellm/llms/azure_text.py:23: in openai_text_completion_config = OpenAITextCompletionConfig() .cache/pypoetry/virtualenvs/astra-assistants-api-eiSmbCzm-py3.10/lib/python3.10/site-packages/litellm/llms/openai.py:192: in __init__ for key, value in locals_.items(): E RuntimeError: dictionary changed size during iteration ``` --- litellm/llms/openai.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/llms/openai.py b/litellm/llms/openai.py index 7acbdfae02e..b8fde6e4f1e 100644 --- a/litellm/llms/openai.py +++ b/litellm/llms/openai.py @@ -96,7 +96,7 @@ class MistralConfig: safe_prompt: Optional[bool] = None, response_format: Optional[dict] = None, ) -> None: - locals_ = locals() + locals_ = locals().copy() for key, value in locals_.items(): if key != "self" and value is not None: setattr(self.__class__, key, value) @@ -211,7 +211,7 @@ class OpenAIConfig: temperature: Optional[int] = None, top_p: Optional[int] = None, ) -> None: - locals_ = locals() + locals_ = locals().copy() for key, value in locals_.items(): if key != "self" and value is not None: setattr(self.__class__, key, value) @@ -294,7 +294,7 @@ class OpenAITextCompletionConfig: temperature: Optional[float] = None, top_p: Optional[float] = None, ) -> None: - locals_ = locals() + locals_ = locals().copy() for key, value in locals_.items(): if key != "self" and value is not None: setattr(self.__class__, key, value)