feature(aws_models): anthropic_version param should be set in constructor and is for bedrock only

Author: drawal1

dsp/modules/aws_models.py

@@ -204,6 +204,9 @@ def __init__(
         self.aws_provider = aws_provider
         self.provider = aws_provider.get_provider_name()
 
+        if isinstance(self.aws_provider, Bedrock):
+            self.kwargs["anthropic_version"] = "bedrock-2023-05-31"
+
         for k, v in kwargs.items():
             self.kwargs[k] = v
 
@@ -233,7 +236,6 @@ def _create_body(self, prompt: str, **kwargs) -> tuple[int, dict[str, str | floa
                 ],
             },
         ]
-        query_args["anthropic_version"] = "bedrock-2023-05-31"
         return (n, query_args)
 
     def _call_model(self, body: str) -> str: