File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -8,13 +8,13 @@ class AWSProvider(ABC):
88 """This abstract class adds support for AWS model providers such as Bedrock and SageMaker.
99 The subclasses such as Bedrock and Sagemaker implement the abstract method _call_model and work in conjunction with the AWSModel classes.
1010 Usage Example:
11- bedrock = Bedrock(region_name="us-west-2")
12- bedrock_mixtral = AWSMistral(bedrock, "mistral.mixtral-8x7b-instruct-v0:1", **kwargs)
13- bedrock_haiku = AWSAnthropic(bedrock, "anthropic.claude-3-haiku-20240307-v1:0", **kwargs)
14- bedrock_llama2 = AWSMeta(bedrock, "meta.llama2-13b-chat-v1", **kwargs)
11+ bedrock = dspy. Bedrock(region_name="us-west-2")
12+ bedrock_mixtral = dspy. AWSMistral(bedrock, "mistral.mixtral-8x7b-instruct-v0:1", **kwargs)
13+ bedrock_haiku = dspy. AWSAnthropic(bedrock, "anthropic.claude-3-haiku-20240307-v1:0", **kwargs)
14+ bedrock_llama2 = dspy. AWSMeta(bedrock, "meta.llama2-13b-chat-v1", **kwargs)
1515
16- sagemaker = Sagemaker(region_name="us-west-2")
17- sagemaker_model = AWSMistral(sagemaker, "<YOUR_ENDPOINT_NAME>", **kwargs)
16+ sagemaker = dspy. Sagemaker(region_name="us-west-2")
17+ sagemaker_model = dspy. AWSMistral(sagemaker, "<YOUR_ENDPOINT_NAME>", **kwargs)
1818 """
1919
2020 def __init__ (
You can’t perform that action at this time.
0 commit comments