-
-
Notifications
You must be signed in to change notification settings - Fork 220
/
oai-code-chat.py
85 lines (67 loc) · 2.13 KB
/
oai-code-chat.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
"""
The most basic use of code-interpreter, using an OpenAssistant agent,
powered by the OpenAI Assistant API's code-interpreter tool.
Run like this:
python3 examples/basic/oai-code-chat.py
"""
import typer
from rich import print
from rich.prompt import Prompt
from dotenv import load_dotenv
import tempfile
from langroid.agent.openai_assistant import (
OpenAIAssistant,
OpenAIAssistantConfig,
AssistantTool,
ToolType,
)
from langroid.parsing.url_loader import URLLoader
from langroid.agent.task import Task
from langroid.language_models.openai_gpt import OpenAIGPTConfig, OpenAIChatModel
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,
)
path = Prompt.ask("Enter a URL or file path, or hit enter if no files")
if path:
# if path is a url, use UrlLoader to get text as a document
if path.startswith("http"):
text = URLLoader([path]).load()[0].content
# save text to a temp file
with tempfile.NamedTemporaryFile(
mode="w", suffix=".txt", delete=False
) as f:
f.write(text)
f.close()
# get the filename
path = f.name
config = OpenAIAssistantConfig(
system_message=sys_msg,
llm=OpenAIGPTConfig(chat_model=OpenAIChatModel.GPT4o),
)
agent = OpenAIAssistant(config)
agent.add_assistant_tools([AssistantTool(type=ToolType.CODE_INTERPRETER)])
if path:
agent.add_assistant_files([path])
task = Task(agent)
task.run(
"""
Help me with some questions,
using the CODE INTERPRETER tool, and any uploaded files as needed.
"""
)
if __name__ == "__main__":
app()