Files
Project-AI-Chronicle/ai-server/backend/ai/embedding.py
T

12 lines
428 B
Python

from langchain_ibm import WatsonxEmbeddings
from backend.config.settings import settings
from langchain_ollama import OllamaEmbeddings
watson_embedding = WatsonxEmbeddings(
model_id="ibm/granite-embedding-278m-multilingual",
url=f"{settings.watsonx_url}",
api_key=f"{settings.watsonx_api_key}",
project_id=f"{settings.watsonx_project_id}",
)
ollama_embedding = OllamaEmbeddings(model="nomic-embed-text-v2-moe")