x

Author: eyurtsev

libs/partners/fireworks/langchain_fireworks/embeddings.py

@@ -4,68 +4,65 @@
 from langchain_core.embeddings import Embeddings
 from langchain_core.pydantic_v1 import BaseModel, Field, SecretStr, root_validator
 from langchain_core.utils import convert_to_secret_str
-
 from openai import OpenAI  # type: ignore
 
 
 class FireworksEmbeddings(BaseModel, Embeddings):
     """Fireworks embedding model integration.
 
-    Setup:
-        Install ``langchain_fireworks`` and set environment variable ``FIREWORKS_API_KEY``.
-
-        .. code-block:: bash
-
-            pip install -U langchain_fireworks
-            export FIREWORKS_API_KEY="your-api-key"
+     Setup:
+         Install ``langchain_fireworks`` and set environment variable ``FIREWORKS_API_KEY``.
 
-    Key init args — completion params:
-        model: str
-            Name of Fireworks model to use.
+         .. code-block:: bash
 
-   Key init args — client params:
-       fireworks_api_key: SecretStr
-           Fireworks API key.
+             pip install -U langchain_fireworks
+             export FIREWORKS_API_KEY="your-api-key"
 
-    See full list of supported init args and their descriptions in the params section.
+     Key init args — completion params:
+         model: str
+             Name of Fireworks model to use.
 
-    Instantiate:
-        .. code-block:: python
+    Key init args — client params:
+        fireworks_api_key: SecretStr
+            Fireworks API key.
 
-            from __module_name__ import FireworksEmbeddings
+     See full list of supported init args and their descriptions in the params section.
 
-            model = FireworksEmbeddings(
-                model='nomic-ai/nomic-embed-text-v1.5'
-                # Use FIREWORKS_API_KEY env var or pass it in directly
-                # fireworks_api_key="..."
-            )
+     Instantiate:
+         .. code-block:: python
 
-    Embed multiple texts:
+             from __module_name__ import FireworksEmbeddings
 
-        .. code-block:: python
+             model = FireworksEmbeddings(
+                 model='nomic-ai/nomic-embed-text-v1.5'
+                 # Use FIREWORKS_API_KEY env var or pass it in directly
+                 # fireworks_api_key="..."
+             )
 
-            vectors = embeddings.embed_documents(['hello', 'goodbye'])
-            # Showing only the first 3 coordinates
-            print(len(vectors))
-            print(vectors[0][:3])
+     Embed multiple texts:
+         .. code-block:: python
 
-        .. code-block:: python
+             vectors = embeddings.embed_documents(['hello', 'goodbye'])
+             # Showing only the first 3 coordinates
+             print(len(vectors))
+             print(vectors[0][:3])
 
-            2
-            [-0.024603435769677162, -0.007543657906353474, 0.0039630369283258915]
+         .. code-block:: python
 
+             2
+             [-0.024603435769677162, -0.007543657906353474, 0.0039630369283258915]
 
-    Embed single text:
 
-        .. code-block:: python
+     Embed single text:
+         .. code-block:: python
 
-            input_text = "The meaning of life is 42"
-            vector = embeddings.embed_query('hello')
-            print(vector[:3])
+             input_text = "The meaning of life is 42"
+             vector = embeddings.embed_query('hello')
+             print(vector[:3])
 
-        .. code-block:: python
+         .. code-block:: python
 
-            [-0.024603435769677162, -0.007543657906353474, 0.0039630369283258915]
+             [-0.024603435769677162, -0.007543657906353474, 0.0039630369283258915]
     """
 
     _client: OpenAI = Field(default=None)