github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-17 23:41:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: update OctoAIEmbeddings to subclass OpenAIEmbeddings (#21805)

Browse Source
This commit is contained in:
Sevin F. Varoglu
2024-05-21 21:29:41 +03:00
committed by GitHub
parent ded53297e0
commit 1bc0ea5496
2 changed files with 74 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
libs/community/tests/integration_tests/embeddings/test_octoai_embeddings.py
View File

@@ -8,27 +8,15 @@ from langchain_community.embeddings.octoai_embeddings import (
def test_octoai_embedding_documents() -> None:
"""Test octoai embeddings."""
documents = ["foo bar"]
embedding = OctoAIEmbeddings(
endpoint_url="<endpoint_url>",
octoai_api_token="<octoai_api_token>",
embed_instruction="Represent this input: ",
query_instruction="Represent this input: ",
model_kwargs=None,
)
embedding = OctoAIEmbeddings()
output = embedding.embed_documents(documents)
assert len(output) == 1
assert len(output[0]) == 768
assert len(output[0]) == 1024
def test_octoai_embedding_query() -> None:
"""Test octoai embeddings."""
document = "foo bar"
embedding = OctoAIEmbeddings(
endpoint_url="<endpoint_url>",
octoai_api_token="<octoai_api_token>",
embed_instruction="Represent this input: ",
query_instruction="Represent this input: ",
model_kwargs=None,
)
embedding = OctoAIEmbeddings()
output = embedding.embed_query(document)
assert len(output) == 768
assert len(output) == 1024