chatbot_001 / chatbot_engine.py
Kanekonkon's picture
Upload folder using huggingface_hub
f947f9c verified
raw
history blame contribute delete
214 Bytes
import langchain
from langchain.chat_models import ChatOpenAI
langchain.verbose = True
def chat(message: str) -> str:
llm = ChatOpenAI(model_name="gpt-4o-mini", temperature=0)
return llm.predict(message)