mirror of
https://github.com/langgenius/dify.git
synced 2026-03-04 15:26:21 +08:00
40 lines
717 B
Python
40 lines
717 B
Python
from decimal import Decimal
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from dify_graph.model_runtime.entities.model_entities import ModelUsage
|
|
|
|
|
|
class EmbeddingUsage(ModelUsage):
|
|
"""
|
|
Model class for embedding usage.
|
|
"""
|
|
|
|
tokens: int
|
|
total_tokens: int
|
|
unit_price: Decimal
|
|
price_unit: Decimal
|
|
total_price: Decimal
|
|
currency: str
|
|
latency: float
|
|
|
|
|
|
class EmbeddingResult(BaseModel):
|
|
"""
|
|
Model class for text embedding result.
|
|
"""
|
|
|
|
model: str
|
|
embeddings: list[list[float]]
|
|
usage: EmbeddingUsage
|
|
|
|
|
|
class FileEmbeddingResult(BaseModel):
|
|
"""
|
|
Model class for file embedding result.
|
|
"""
|
|
|
|
model: str
|
|
embeddings: list[list[float]]
|
|
usage: EmbeddingUsage
|