Skip to content

Commit 4326033

Browse files
Update aws_providers.py
1 parent 87540da commit 4326033

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

dsp/modules/aws_providers.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff 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__(

0 commit comments

Comments
 (0)