13 lines
341 B
Python
13 lines
341 B
Python
from langchain_ibm import ChatWatsonx
|
|
from backend.config.settings import settings
|
|
|
|
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,
|
|
params={
|
|
"temperature": 0
|
|
}
|
|
) |