/
githubmirror
/
langroid
Обзор
Документация
Войти
/
githubmirror
/
langroid
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
examples/basic/oai-asst-chat.py
54 строки
1 KB
Prasad Chalasani
Add TaskTool dynamic sub-agent spawn example + ruff auto-fix for examples (#876)
28 июн 2025, 00:02
Не верифицирован
28 июн 2025, 00:02
7169fc4
Код
Авторство
О чём код?
""" The most basic chatbot example, using an OpenAssistant agent, powered by the OpenAI Assistant API. Run like this: python3 examples/basic/oai-asst-chat.py """ import typer from dotenv import load_dotenv from rich import print from rich.prompt import Prompt from langroid.agent.openai_assistant import OpenAIAssistant, OpenAIAssistantConfig from langroid.agent.task import Task from langroid.language_models.openai_gpt import OpenAIChatModel, OpenAIGPTConfig from langroid.utils.logging import setup_colored_logging app = typer.Typer() setup_colored_logging() @app.command() def chat() -> None: print( """ [blue]Welcome to the basic chatbot! Enter x or q to quit at any point. """ ) load_dotenv() default_sys_msg = "You are a helpful assistant. Be concise in your answers." sys_msg = Prompt.ask( "[blue]Tell me who I am. Hit Enter for default, or type your own\n", default=default_sys_msg, ) config = OpenAIAssistantConfig( system_message=sys_msg, llm=OpenAIGPTConfig(chat_model=OpenAIChatModel.GPT4o), ) agent = OpenAIAssistant(config) task = Task(agent) task.run() if __name__ == "__main__": app()