chatlas 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

chatlas/types/anthropic/_submit.py

@@ -18,7 +18,6 @@ class SubmitInputArgs(TypedDict, total=False):
     max_tokens: int
     messages: Iterable[anthropic.types.message_param.MessageParam]
     model: Union[
-        str,
         Literal[
             "claude-3-5-haiku-latest",
             "claude-3-5-haiku-20241022",
@@ -32,6 +31,7 @@ class SubmitInputArgs(TypedDict, total=False):
             "claude-2.1",
             "claude-2.0",
         ],
+        str,
     ]
     stop_sequences: Union[list[str], anthropic.NotGiven]
     stream: Union[Literal[False], Literal[True], anthropic.NotGiven]

chatlas/types/google/_client.py

@@ -3,99 +3,20 @@
 # ---------------------------------------------------------
 
 
-from typing import Any, Callable, Iterable, TypedDict, Union
+from typing import Optional, TypedDict, Union
 
-import google.ai.generativelanguage_v1beta.types.content
-import google.ai.generativelanguage_v1beta.types.file
-import google.ai.generativelanguage_v1beta.types.generative_service
-import google.generativeai.types.content_types
-import google.generativeai.types.file_types
-import google.generativeai.types.generation_types
+import google.auth.credentials
+import google.genai.client
+import google.genai.types
 
 
 class ChatClientArgs(TypedDict, total=False):
-    model_name: str
-    generation_config: Union[
-        google.ai.generativelanguage_v1beta.types.generative_service.GenerationConfig,
-        google.generativeai.types.generation_types.GenerationConfigDict,
-        google.generativeai.types.generation_types.GenerationConfig,
-        None,
-    ]
-    tools: Union[
-        google.generativeai.types.content_types.FunctionLibrary,
-        Iterable[
-            Union[
-                str,
-                google.generativeai.types.content_types.Tool,
-                google.ai.generativelanguage_v1beta.types.content.Tool,
-                google.generativeai.types.content_types.ToolDict,
-                Iterable[
-                    Union[
-                        google.generativeai.types.content_types.FunctionDeclaration,
-                        google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                        dict[str, Any],
-                        Callable[..., Any],
-                    ]
-                ],
-                google.generativeai.types.content_types.FunctionDeclaration,
-                google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                dict[str, Any],
-                Callable[..., Any],
-            ]
-        ],
-        str,
-        google.generativeai.types.content_types.Tool,
-        google.ai.generativelanguage_v1beta.types.content.Tool,
-        google.generativeai.types.content_types.ToolDict,
-        Iterable[
-            Union[
-                google.generativeai.types.content_types.FunctionDeclaration,
-                google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                dict[str, Any],
-                Callable[..., Any],
-            ]
-        ],
-        google.generativeai.types.content_types.FunctionDeclaration,
-        google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-        dict[str, Any],
-        Callable[..., Any],
-        None,
-    ]
-    tool_config: Union[
-        google.generativeai.types.content_types.ToolConfigDict,
-        google.ai.generativelanguage_v1beta.types.content.ToolConfig,
-        None,
-    ]
-    system_instruction: Union[
-        google.ai.generativelanguage_v1beta.types.content.Content,
-        google.generativeai.types.content_types.ContentDict,
-        Iterable[
-            Union[
-                google.ai.generativelanguage_v1beta.types.content.Part,
-                google.generativeai.types.content_types.PartDict,
-                google.ai.generativelanguage_v1beta.types.content.Blob,
-                google.generativeai.types.content_types.BlobDict,
-                Any,
-                str,
-                google.ai.generativelanguage_v1beta.types.content.FunctionCall,
-                google.ai.generativelanguage_v1beta.types.content.FunctionResponse,
-                google.generativeai.types.file_types.FileDataDict,
-                google.ai.generativelanguage_v1beta.types.content.FileData,
-                google.ai.generativelanguage_v1beta.types.file.File,
-                google.generativeai.types.file_types.File,
-            ]
-        ],
-        google.ai.generativelanguage_v1beta.types.content.Part,
-        google.generativeai.types.content_types.PartDict,
-        google.ai.generativelanguage_v1beta.types.content.Blob,
-        google.generativeai.types.content_types.BlobDict,
-        Any,
-        str,
-        google.ai.generativelanguage_v1beta.types.content.FunctionCall,
-        google.ai.generativelanguage_v1beta.types.content.FunctionResponse,
-        google.generativeai.types.file_types.FileDataDict,
-        google.ai.generativelanguage_v1beta.types.content.FileData,
-        google.ai.generativelanguage_v1beta.types.file.File,
-        google.generativeai.types.file_types.File,
-        None,
+    vertexai: Optional[bool]
+    api_key: Optional[str]
+    credentials: Optional[google.auth.credentials.Credentials]
+    project: Optional[str]
+    location: Optional[str]
+    debug_config: Optional[google.genai.client.DebugConfig]
+    http_options: Union[
+        google.genai.types.HttpOptions, google.genai.types.HttpOptionsDict, None
     ]

chatlas/types/google/_submit.py

@@ -3,111 +3,64 @@
 # ---------------------------------------------------------
 
 
-from typing import Any, Callable, Iterable, TypedDict, Union
+from typing import TypedDict, Union
 
-import google.ai.generativelanguage_v1beta.types.content
-import google.ai.generativelanguage_v1beta.types.file
-import google.ai.generativelanguage_v1beta.types.generative_service
-import google.generativeai.types.content_types
-import google.generativeai.types.file_types
-import google.generativeai.types.generation_types
-import google.generativeai.types.helper_types
+import google.genai.types
+import PIL.Image
 
 
 class SubmitInputArgs(TypedDict, total=False):
+    model: str
     contents: Union[
-        google.ai.generativelanguage_v1beta.types.content.Content,
-        google.generativeai.types.content_types.ContentDict,
-        Iterable[
+        list[
             Union[
-                google.ai.generativelanguage_v1beta.types.content.Part,
-                google.generativeai.types.content_types.PartDict,
-                google.ai.generativelanguage_v1beta.types.content.Blob,
-                google.generativeai.types.content_types.BlobDict,
-                Any,
+                google.genai.types.Content,
+                list[
+                    Union[
+                        google.genai.types.File,
+                        google.genai.types.Part,
+                        PIL.Image.Image,
+                        str,
+                    ]
+                ],
+                google.genai.types.File,
+                google.genai.types.Part,
+                PIL.Image.Image,
                 str,
-                google.ai.generativelanguage_v1beta.types.content.FunctionCall,
-                google.ai.generativelanguage_v1beta.types.content.FunctionResponse,
-                google.generativeai.types.file_types.FileDataDict,
-                google.ai.generativelanguage_v1beta.types.content.FileData,
-                google.ai.generativelanguage_v1beta.types.file.File,
-                google.generativeai.types.file_types.File,
             ]
         ],
-        google.ai.generativelanguage_v1beta.types.content.Part,
-        google.generativeai.types.content_types.PartDict,
-        google.ai.generativelanguage_v1beta.types.content.Blob,
-        google.generativeai.types.content_types.BlobDict,
-        Any,
-        str,
-        google.ai.generativelanguage_v1beta.types.content.FunctionCall,
-        google.ai.generativelanguage_v1beta.types.content.FunctionResponse,
-        google.generativeai.types.file_types.FileDataDict,
-        google.ai.generativelanguage_v1beta.types.content.FileData,
-        google.ai.generativelanguage_v1beta.types.file.File,
-        google.generativeai.types.file_types.File,
-        Iterable[
+        google.genai.types.Content,
+        list[
             Union[
-                google.ai.generativelanguage_v1beta.types.content.Content,
-                google.generativeai.types.content_types.ContentDict,
+                google.genai.types.File, google.genai.types.Part, PIL.Image.Image, str
             ]
         ],
-        None,
-    ]
-    generation_config: Union[
-        google.ai.generativelanguage_v1beta.types.generative_service.GenerationConfig,
-        google.generativeai.types.generation_types.GenerationConfigDict,
-        google.generativeai.types.generation_types.GenerationConfig,
-        None,
-    ]
-    stream: bool
-    tools: Union[
-        google.generativeai.types.content_types.FunctionLibrary,
-        Iterable[
+        google.genai.types.File,
+        google.genai.types.Part,
+        PIL.Image.Image,
+        str,
+        list[
             Union[
-                str,
-                google.generativeai.types.content_types.Tool,
-                google.ai.generativelanguage_v1beta.types.content.Tool,
-                google.generativeai.types.content_types.ToolDict,
-                Iterable[
+                google.genai.types.Content,
+                list[
                     Union[
-                        google.generativeai.types.content_types.FunctionDeclaration,
-                        google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                        dict[str, Any],
-                        Callable[..., Any],
+                        google.genai.types.File,
+                        google.genai.types.Part,
+                        PIL.Image.Image,
+                        str,
                     ]
                 ],
-                google.generativeai.types.content_types.FunctionDeclaration,
-                google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                dict[str, Any],
-                Callable[..., Any],
-            ]
-        ],
-        str,
-        google.generativeai.types.content_types.Tool,
-        google.ai.generativelanguage_v1beta.types.content.Tool,
-        google.generativeai.types.content_types.ToolDict,
-        Iterable[
-            Union[
-                google.generativeai.types.content_types.FunctionDeclaration,
-                google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-                dict[str, Any],
-                Callable[..., Any],
+                google.genai.types.File,
+                google.genai.types.Part,
+                PIL.Image.Image,
+                str,
+                google.genai.types.ContentDict,
             ]
         ],
-        google.generativeai.types.content_types.FunctionDeclaration,
-        google.ai.generativelanguage_v1beta.types.content.FunctionDeclaration,
-        dict[str, Any],
-        Callable[..., Any],
-        None,
-    ]
-    tool_config: Union[
-        google.generativeai.types.content_types.ToolConfigDict,
-        google.ai.generativelanguage_v1beta.types.content.ToolConfig,
-        None,
+        google.genai.types.ContentDict,
     ]
-    request_options: Union[
-        google.generativeai.types.helper_types.RequestOptions,
-        google.generativeai.types.helper_types.RequestOptionsDict,
+    config: Union[
+        google.genai.types.GenerateContentConfig,
+        google.genai.types.GenerateContentConfigDict,
         None,
     ]

chatlas/types/openai/_submit.py

@@ -38,6 +38,8 @@ class SubmitInputArgs(TypedDict, total=False):
     model: Union[
         str,
         Literal[
+            "o3-mini",
+            "o3-mini-2025-01-31",
             "o1",
             "o1-2024-12-17",
             "o1-preview",
@@ -105,7 +107,7 @@ class SubmitInputArgs(TypedDict, total=False):
         openai.NotGiven,
     ]
     presence_penalty: Union[float, None, openai.NotGiven]
-    reasoning_effort: Union[Literal["low", "medium", "high"], openai.NotGiven]
+    reasoning_effort: Union[Literal["low", "medium", "high"], None, openai.NotGiven]
     response_format: Union[
         openai.types.shared_params.response_format_text.ResponseFormatText,
         openai.types.shared_params.response_format_json_object.ResponseFormatJSONObject,

{chatlas-0.3.0.dist-info → chatlas-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chatlas
-Version: 0.3.0
+Version: 0.4.0
 Summary: A simple and consistent interface for chatting with LLMs
 Project-URL: Homepage, https://posit-dev.github.io/chatlas
 Project-URL: Documentation, https://posit-dev.github.io/chatlas
@@ -22,7 +22,7 @@ Requires-Dist: pydantic>=2.0
 Requires-Dist: rich
 Provides-Extra: dev
 Requires-Dist: anthropic[bedrock]; extra == 'dev'
-Requires-Dist: google-generativeai>=0.8.3; extra == 'dev'
+Requires-Dist: google-genai>=1.2.0; extra == 'dev'
 Requires-Dist: matplotlib; extra == 'dev'
 Requires-Dist: numpy>1.24.4; extra == 'dev'
 Requires-Dist: openai; extra == 'dev'
@@ -49,7 +49,9 @@ Requires-Dist: pytest>=8.3.2; extra == 'test'
 Requires-Dist: syrupy>=4; extra == 'test'
 Description-Content-Type: text/markdown
 
-# chatlas
+<h1 class="unnumbered unlisted"> chatlas <a href="https://posit-dev.github.io/chatlas"><img src="docs/images/logo.png" align="right" height="138" alt="chatlas website" /></a> </h1> 
+
+
 
 <p>
 <!-- badges start -->
@@ -60,11 +62,11 @@ Description-Content-Type: text/markdown
 </p>
 
 chatlas provides a simple and unified interface across large language model (llm) providers in Python. 
-It abstracts away complexity from common tasks like streaming chat interfaces, tool calling, structured output, and much more.
-chatlas helps you prototype faster without painting you into a corner; for example, switching providers is as easy as changing one line of code, but provider specific features are still accessible when needed.
-Developer experience is also a key focus of chatlas: typing support, rich console output, and built-in tooling are all included.
+It helps you prototype faster by abstracting away complexity from common tasks like streaming chat interfaces, tool calling, structured output, and much more.
+Switching providers is also as easy as changing one line of code, but you can also reach for provider-specific features when you need them.
+Developer experience is also a key focus of chatlas: typing support, rich console output, and extension points are all included.
 
-(Looking for something similar to chatlas, but in R? Check out [elmer](https://elmer.tidyverse.org/)!)
+(Looking for something similar to chatlas, but in R? Check out [ellmer](https://ellmer.tidyverse.org/)!)
 
 ## Install
 
@@ -96,6 +98,7 @@ It also supports the following enterprise cloud providers:
 
 * AWS Bedrock: [`ChatBedrockAnthropic()`](https://posit-dev.github.io/chatlas/reference/ChatBedrockAnthropic.html).
 * Azure OpenAI: [`ChatAzureOpenAI()`](https://posit-dev.github.io/chatlas/reference/ChatAzureOpenAI.html).
+* Vertex AI: [`ChatVertex()`](https://posit-dev.github.io/chatlas/reference/ChatVertex.html).
 
 To use a model provider that isn't listed here, you have two options:
 

{chatlas-0.3.0.dist-info → chatlas-0.4.0.dist-info}/RECORD

@@ -1,37 +1,38 @@
-chatlas/__init__.py,sha256=OJbTO71ne1O9SDxkwIKOMpCMKbh0T8eDpYPFhrAb28A,974
-chatlas/_anthropic.py,sha256=ssv6k-XZetwv1hNk7Qs8dAWlcPvElzxjrG0Fh2w50xE,24373
-chatlas/_chat.py,sha256=p_MS1LQ-EBlvCzGTbjY5Bdlt1meDsfSbP517blkwSjY,44774
-chatlas/_content.py,sha256=vpWF_WKS2tCDUtnL8l9lfW6b6g9e7LbDKP-_TegauVE,5883
+chatlas/__init__.py,sha256=HtvcxTy7A3CO7d6JSZIGDqIPGBRuo7Vte11nazn_VNE,1004
+chatlas/_anthropic.py,sha256=fygtGcKi5qM1w40AZbcnJcsPUkBEyWuqpoyCOKG-kOc,24259
+chatlas/_chat.py,sha256=15j4w1cvFXrqsf7sGMv8B_TidEiow-szbRr6G-HiSqM,45990
+chatlas/_content.py,sha256=79FTmKrByS-rlg6XTbKQh4klaA4O3MJiljMrYrkr1mU,6334
 chatlas/_content_image.py,sha256=4nk9wTvLtNmtcytdFp8p9otEV5-0_K6wzIxCyK0PIEI,8367
 chatlas/_display.py,sha256=_IcQcvpyTNjGHOpY70_LOrDWwTjzdkziy6pTvxHEiWI,4053
 chatlas/_github.py,sha256=D3L7Qu35K-M1qEW7-w-Oq-pF-9mVetia3MHYNNLEYtU,4373
-chatlas/_google.py,sha256=jfBp_C_qjbvs48QPE_ykSsfRhjz7htb-G389qLxvSC4,15279
+chatlas/_google.py,sha256=RUXKUChiEqLzhIO3aWQn2dooJEmHNaX8gFK1a5Pzw1Q,18763
 chatlas/_groq.py,sha256=3VnYiKdxJTHPhEgUKnL2nY5uYL2L4PKBo7GZMwR0D8k,4158
 chatlas/_interpolate.py,sha256=ykwLP3x-ya9Q33U4knSU75dtk6pzJAeythEEIW-43Pc,3631
 chatlas/_logging.py,sha256=7a20sAl1PkW1qBNrfd_ieUbQXV8Gf4Vuf0Wn62LNBmk,2290
-chatlas/_merge.py,sha256=Xt2uutLdEmYAGfGCa8GCEd8sdNadQM5o3l-zuIQFbWU,3923
+chatlas/_merge.py,sha256=SGj_BetgA7gaOqSBKOhYmW3CYeQKTEehFrXvx3y4OYE,3924
 chatlas/_ollama.py,sha256=ze-RoHEbf62dYmXDDKjNGqaEZaKCZdcBEyFwQMDQxkQ,3760
-chatlas/_openai.py,sha256=nJgp3JJlpxeXn_dhIrXNffQk9vh3ycYy1D2fNip0qEs,24004
+chatlas/_openai.py,sha256=Z1y_NdWr3TyrrFGtvYEaTU7cdMQAqLzMiK67-kIa8vM,23852
 chatlas/_perplexity.py,sha256=Bw_mlM8N8egGKIrbNerTn2pMlybugADOshjYOfN1ixM,4446
-chatlas/_provider.py,sha256=i16I2hkBat1fYEMcFsU0gYNr6Tcg8zJ2xHDnEY7WRY4,4009
+chatlas/_provider.py,sha256=YmdBbz_u5aP_kBxl6s26OPiSnWG_vZ_fvf9L2qvBmyI,3809
 chatlas/_tokens.py,sha256=3W3EPUp9eWXUiwuzJwEPBv43AUznbK46pm59Htti7z4,2392
 chatlas/_tokens_old.py,sha256=L9d9oafrXvEx2u4nIn_Jjn7adnQyLBnYBuPwJUE8Pl8,5005
 chatlas/_tools.py,sha256=-qt4U1AFkebQoX9kpsBy5QXK8a2PpHX6Amgm44gcQ68,4113
 chatlas/_turn.py,sha256=nKwk20FrOIrZX4xJxdGyUEpwUH2H-UYcoJLlO2ZD5iU,4836
 chatlas/_typing_extensions.py,sha256=YdzmlyPSBpIEcsOkoz12e6jETT1XEMV2Q72haE4cfwY,1036
 chatlas/_utils.py,sha256=qAiWuDx-uG8BGFZ_PWvum9wpN-WogdItO32X4pRhhLs,2762
-chatlas/types/__init__.py,sha256=pgHl8pd2Ytskd6lkfNtm98Yj1ZP0b3R35RH4Uht2BAs,694
+chatlas/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+chatlas/types/__init__.py,sha256=P_EDL4eqsigKwB-u2qRmKlYQS5Y65m7oWjGC3cYmxO4,719
 chatlas/types/anthropic/__init__.py,sha256=OwubA-DPHYpYo0XyRyAFwftOI0mOxtHzAyhUSLcDx54,417
 chatlas/types/anthropic/_client.py,sha256=G0LRhoFBcsSOMr5qhP-0rAScsVXaVlHCpggfVp54bnQ,690
 chatlas/types/anthropic/_client_bedrock.py,sha256=mNazQlu0pQt8JdzrYn3LKNgE4n732GjhQUJdQQK9QkY,785
-chatlas/types/anthropic/_submit.py,sha256=BhkJu0OhIMiD3qiUNqJ4litBQCW-7KRr_YZp9Xr9KGA,2091
+chatlas/types/anthropic/_submit.py,sha256=cSSomSJ2zfAHu8ilzMN3XxRTLANh00oK_t5kH6N2lCI,2091
 chatlas/types/google/__init__.py,sha256=ZJhi8Kwvio2zp8T1TQqmvdHqkS-Khb6BGESPjREADgo,337
-chatlas/types/google/_client.py,sha256=YA5hsT-m-KcONKtwpCULYMnGwMPfkScpvhjx_qBLg5o,4421
-chatlas/types/google/_submit.py,sha256=yp1wtp5eScLlHDNxeXl0qJOKv7SWLnRQ8oslupRFUBE,4839
+chatlas/types/google/_client.py,sha256=t7aKbxYq_xOA1Z3RnWcjewifdQFSHi7vKEj6MyKMCJk,729
+chatlas/types/google/_submit.py,sha256=b-ZqMvI551Ia7pFlWdqUQJjov3neHmVwLFw-P2bgU8w,1883
 chatlas/types/openai/__init__.py,sha256=Q2RAr1bSH1nHsxICK05nAmKmxdhKmhbBkWD_XHiVSrI,411
 chatlas/types/openai/_client.py,sha256=YGm_EHtRSSHeeOZe-CV7oNvMJpEblEta3UTuU7lSRO8,754
 chatlas/types/openai/_client_azure.py,sha256=jx8D_p46CLDGzTP-k-TtGzj-f3junj6or-86m8DD_0w,858
-chatlas/types/openai/_submit.py,sha256=f2o3rNJZcWBhwjomXrs3Mh0V_4vNa9N5eEBkMvJDofQ,6028
-chatlas-0.3.0.dist-info/METADATA,sha256=eIPyvdcXpYN4s6txY5GMgBttUsmS5868D10U5zzTchs,13164
-chatlas-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-chatlas-0.3.0.dist-info/RECORD,,
+chatlas/types/openai/_submit.py,sha256=VX2D1eQHF19XRIt-oq90cBSw_3ChokGpwVG7-b2AJOU,6091
+chatlas-0.4.0.dist-info/METADATA,sha256=-ayykYk3sVvvPRlW6jj15U6S76ywblY0p1sD8trc640,13389
+chatlas-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+chatlas-0.4.0.dist-info/RECORD,,