Merge pull request #993 from 4daJKong/patch-1

arnavsinghvi11 · web-flow · commit 012b07ac221e · 2024-05-11T11:52:38.000-07:00
Update anthropic.py
diff --git a/dsp/modules/anthropic.py b/dsp/modules/anthropic.py
@@ -62,7 +62,7 @@ def __init__(
         }
         self.kwargs["model"] = model
         self.history: list[dict[str, Any]] = []
-        self.client = Anthropic(api_key=api_key)
+        self.client = Anthropic(api_key=api_key, base_url=api_base)
 
     def log_usage(self, response):
         """Log the total tokens from the Anthropic API response."""

Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,7 @@ def __init__(`
`62`	`62`	`}`
`63`	`63`	`self.kwargs["model"] = model`
`64`	`64`	`self.history: list[dict[str, Any]] = []`
`65`		`- self.client = Anthropic(api_key=api_key)`
	`65`	`+ self.client = Anthropic(api_key=api_key, base_url=api_base)`
`66`	`66`
`67`	`67`	`def log_usage(self, response):`
`68`	`68`	`"""Log the total tokens from the Anthropic API response."""`