Gen_AI_Test / backend /llm.py
AyushMatchbest
initial
f9b7370
Raw
History Blame Contribute Delete
337 Bytes
from langchain_openai import ChatOpenAI
import os
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
if not OPENAI_API_KEY:
raise RuntimeError("OPENAI_API_KEY is not set")
llm = ChatOpenAI(base_url="https://api.a4f.co/v1", model="provider-5/gpt-4o-mini")
def chatgpt(input):
response = llm.invoke(input)
return response.content