nanfangwuyu21's picture
Update the chatbot to an LangGraph agent, handling four tasks including RAG, summary, translate and email. Update prompts. Update Gradio UI.
6861e2a
raw
history blame contribute delete
467 Bytes
from langchain_huggingface import HuggingFaceEmbeddings
from app.models.AzureOpenAI import azure_llm
from app.models.OpenAI import llm as openai_llm
from app.config import model
# model = TinyLlamaModel()
# LLM =
LLM = openai_llm if model == "OpenAI" else azure_llm
Embedding_model_en = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2") # only English
Embedding_model_de = HuggingFaceEmbeddings(model_name="danielheinz/e5-base-sts-en-de")