Adding AntrhopicLLM

parkervg · Sep 1, 2024 · 308dab9 · 308dab9
1 parent 4c93ce1
commit 308dab9
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 25 deletions.
diff --git a/blendsql/models/remote/_anthropic.py b/blendsql/models/remote/_anthropic.py
@@ -10,6 +10,31 @@
 
 
 class AnthropicLLM(RemoteModel):
+    """Class for Anthropic Model API.
+
+    Args:
+        model_name_or_path: Name of the Anthropic model to use
+        env: Path to directory of .env file, or to the file itself to load as a dotfile.
+            Should contain the variable `ANTRHOPIC_API_KEY`
+        config: Optional argument mapping to use in loading model
+        caching: Bool determining whether we access the model's cache
+
+    Examples:
+        Given the following `.env` file in the directory above current:
+        ```text
+        ANTRHOPIC_API_KEY=my_api_key
+        ```
+        ```python
+        from blendsql.models import AnthropicLLM
+
+        model = AnthropicLLM(
+            "claude-3-5-sonnet-20240620",
+            env="..",
+            config={"temperature": 0.7}
+        )
+        ```
+    """
+
     def __init__(
         self,
         model_name_or_path: str,
@@ -18,30 +43,6 @@ def __init__(
         caching: bool = True,
         **kwargs,
     ):
-        """Class for Anthropic Model API.
-
-        Args:
-            model_name_or_path: Name of the Anthropic model to use
-            env: Path to directory of .env file, or to the file itself to load as a dotfile.
-                Should contain the variable `ANTRHOPIC_API_KEY`
-            config: Optional argument mapping to use in loading model
-            caching: Bool determining whether we access the model's cache
-
-        Examples:
-            Given the following `.env` file in the directory above current:
-            ```text
-            ANTRHOPIC_API_KEY=my_api_key
-            ```
-            ```python
-            from blendsql.models import AnthropicLLM
-
-            model = AnthropicLLM(
-                "claude-3-5-sonnet-20240620",
-                env="..",
-                config={"temperature": 0.7}
-            )
-            ```
-        """
         if not _has_anthropic:
             raise ImportError(
                 "Please install anthropic with `pip install anthropic`!"

diff --git a/docs/reference/blenders/anthropic.md b/docs/reference/blenders/anthropic.md
@@ -10,6 +10,6 @@ hide:
 
 ## AnthropicLLM
 
-::: blendsql.models.local._anthropic.AnthropicLLM
+::: blendsql.models.remote._anthropic.AnthropicLLM
     handler: python
     show_source: false