File size: 420 Bytes
85e517e
3b9afc9
 
130d92d
85e517e
3b9afc9
 
a66da40
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

model = HfApiModel()
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()