Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pass in direction

Browse Source
This commit is contained in:
goulustis
2025-10-13 21:44:15 +08:00
parent c13984e042
commit 76e7668ab0
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lang_agent/rag/emb.py
View File

@@ -15,7 +15,7 @@ class QwenEmbeddings(Embeddings):
model: str = "text-embedding-v4", model: str = "text-embedding-v4",
max_workers: int = 5, max_workers: int = 5,
embedding_dimension: int = 512, embedding_dimension: int = 512,
batch_size: int = 10, # DashScope supports up to 10 texts per batch batch_size: int = 10, # NOTE: DashScope supports up to 10 texts per batch
rate_limit_delay: float = 0.00001): rate_limit_delay: float = 0.00001):
""" """
Initialize Qwen embeddings Initialize Qwen embeddings
@@ -45,7 +45,8 @@ class QwenEmbeddings(Embeddings):
# DashScope supports batch processing natively # DashScope supports batch processing natively
response = TextEmbedding.call( response = TextEmbedding.call(
model=self.model, model=self.model,
input=texts # Pass list directly input=texts, # Pass list directly
dimension=self.embedding_dimension
) )
if response.status_code == 200: if response.status_code == 200: