AI 및 fastAPI 초기 구조 셋팅

This commit is contained in:
2026-06-22 16:14:02 +09:00
parent 37133ef694
commit 3a76b24df0
10 changed files with 106 additions and 0 deletions
+9
View File
@@ -0,0 +1,9 @@
from langchain_ibm import WatsonxEmbeddings
from backend.config.settings import settings
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}",
)
+17
View File
@@ -0,0 +1,17 @@
from langchain_ibm import ChatWatsonx
from backend.config.settings import settings
from langchain_openai import ChatOpenAI
watson_llm = ChatWatsonx(
model_id="ibm/granite-4-h-small",
url=f"{settings.watsonx_url}",
api_key=f"{settings.watsonx_api_key}",
project_id=f"{settings.watsonx_project_id}",
max_tokens=2000,
)
hugging_llm = ChatOpenAI(
base_url="https://router.huggingface.co/v1",
api_key=f"{settings.hf_token}",
model_name="Qwen/Qwen3-8B:nscale",
)