File size: 296 Bytes
565e754
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from langchain_openai import ChatOpenAI


def get_model(api_key: str, model: str):
    llm = ChatOpenAI(
        model=model,
        max_retries=2,
        openai_api_key=api_key,
        openai_api_base="https://api.proxyapi.ru/openrouter/v1",
    )
    llm.verbose = False
    
    return llm