Add discriminator (api_type) to AnyLLMConfig to improve integration with Pydantic Settings (#5)

Author: vrslev

any_llm_client/main.py

@@ -1,14 +1,16 @@
 import functools
 import typing
 
+import pydantic
+
 from any_llm_client.clients.mock import MockLLMClient, MockLLMConfig
 from any_llm_client.clients.openai import OpenAIClient, OpenAIConfig
 from any_llm_client.clients.yandexgpt import YandexGPTClient, YandexGPTConfig
 from any_llm_client.core import LLMClient
 from any_llm_client.retry import RequestRetryConfig
 
 
-AnyLLMConfig = YandexGPTConfig | OpenAIConfig | MockLLMConfig
+AnyLLMConfig = typing.Annotated[YandexGPTConfig | OpenAIConfig | MockLLMConfig, pydantic.Discriminator("api_type")]
 
 
 if typing.TYPE_CHECKING: