File size: 400 Bytes
85e517e
3b9afc9
 
130d92d
85e517e
3b9afc9
 
130d92d
 
3b9afc9
 
130d92d
3b9afc9
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import gradio as gr
import random
from smolagents import GradioUI, CodeAgent, HfApiModel


from retriever import load_guest_dataset


guest_info_tool = load_guest_dataset()

alfred = CodeAgent(
        tools=[guest_info_tool],
        model=model,
        add_base_tools=True,
        planning_interval=3   # Enable planning every 3 steps
)
 
if __name__ == "__main__":
    GradioUI(alfred).launch()