cohere-ai / cohere-python

Python Library for Accessing the Cohere API
https://docs.cohere.ai
MIT License
278 stars 60 forks source link

Move models param to the endpoint rather than constructor #516

Closed billytrend-cohere closed 2 months ago

billytrend-cohere commented 2 months ago

This PR removes the optional chat_model, embed_model, and generate_model parameters from the __init__ method of the AwsClient class and its subclasses BedrockClient and SagemakerClient. It also updates the test cases in tests/test_aws_client.py to use the models dictionary instead of hardcoding the model values.

Summary of Changes