File size: 6,608 Bytes
6bec1f5
83d3ac3
 
6bec1f5
 
 
 
 
 
 
ac62b55
6bec1f5
 
16d0da9
 
dbaa2bd
83d3ac3
57ee2e6
6bec1f5
57ee2e6
6bec1f5
dbaa2bd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ac62b55
dbaa2bd
 
 
6bec1f5
dbaa2bd
6bec1f5
dbaa2bd
 
6bec1f5
dbaa2bd
 
6bec1f5
dbaa2bd
 
6bec1f5
dbaa2bd
6bec1f5
dbaa2bd
 
 
6bec1f5
dbaa2bd
 
 
6bec1f5
dbaa2bd
ac62b55
dbaa2bd
 
 
6bec1f5
 
 
 
ac62b55
 
 
83d3ac3
6bec1f5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
dbaa2bd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6bec1f5
b29bac3
ac62b55
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
import start
import gradio as gr
import pandas as pd
from glob import glob
from pathlib import Path
from tabs.dashboard import df
from tabs.faq import (
    about_olas_predict_benchmark,
    about_olas_predict,
    about_the_dataset,
    about_the_tools,
)
from tabs.howto_benchmark import how_to_run

# Feature temporarily disabled til HF support helps us with the Space Error
from tabs.run_benchmark import run_benchmark_main


demo = gr.Blocks()


def run_benchmark_gradio(
    tool_name,
    model_name,
    num_questions,
    openai_api_key,
    anthropic_api_key,
    openrouter_api_key,
):
    """Run the benchmark using inputs."""
    if tool_name is None:
        return "Please enter the name of your tool."
    if (
        openai_api_key is None
        and anthropic_api_key is None
        and openrouter_api_key is None
    ):
        return "Please enter either OpenAI or Anthropic or OpenRouter API key."

    result = run_benchmark_main(
        tool_name,
        model_name,
        num_questions,
        openai_api_key,
        anthropic_api_key,
        openrouter_api_key,
    )

    if result == "completed":
        # get the results file in the results directory
        fns = glob("results/*.csv")

        print(f"Number of files in results directory: {len(fns)}")

        # convert to Path
        files = [Path(file) for file in fns]

        # get results and summary files
        results_files = [file for file in files if "results" in file.name]

        # the other file is the summary file
        summary_files = [file for file in files if "summary" in file.name]

        print(results_files, summary_files)

        # get the path with results
        results_df = pd.read_csv(results_files[0])
        summary_df = pd.read_csv(summary_files[0])

        # make sure all df float values are rounded to 4 decimal places
        results_df = results_df.round(4)
        summary_df = summary_df.round(4)

        return gr.Dataframe(value=results_df), gr.Dataframe(value=summary_df)

    return gr.Textbox(
        label="Benchmark Result", value=result, interactive=False
    ), gr.Textbox(label="Summary", value="")


with demo:
    gr.HTML("<h1>Olas Predict Benchmark</hjson>")
    gr.Markdown(
        "Leaderboard showing the performance of Olas Predict tools on the Autocast dataset and overview of the project."
    )

    with gr.Tabs() as tabs:
        # first tab - leaderboard
        with gr.TabItem("🏅 Benchmark Leaderboard", id=0):

            gr.components.Dataframe(
                value=df,
            )

        # second tab - about
        with gr.TabItem("ℹ️ About"):
            with gr.Row():
                with gr.Accordion("About the Benchmark", open=False):
                    gr.Markdown(about_olas_predict_benchmark)
            with gr.Row():
                with gr.Accordion("About the Tools", open=False):
                    gr.Markdown(about_the_tools)
            with gr.Row():
                with gr.Accordion("About the Autocast Dataset", open=False):
                    gr.Markdown(about_the_dataset)
            with gr.Row():
                with gr.Accordion("About Olas", open=False):
                    gr.Markdown(about_olas_predict)

        # third tab - how to run the benchmark
        with gr.TabItem("🚀 Contribute"):
            gr.Markdown(how_to_run)

        # fourth tab - run the benchmark
        with gr.TabItem("🔥 Run the Benchmark"):
            with gr.Row():
                tool_name = gr.Dropdown(
                    [
                        "prediction-offline",
                        "prediction-online",
                        # "prediction-online-summarized-info",
                        # "prediction-offline-sme",
                        # "prediction-online-sme",
                        "prediction-request-rag",
                        "prediction-request-reasoning",
                        # "prediction-url-cot-claude",
                        # "prediction-request-rag-cohere",
                        # "prediction-with-research-conservative",
                        # "prediction-with-research-bold",
                    ],
                    label="Tool Name",
                    info="Choose the tool to run",
                )
                model_name = gr.Dropdown(
                    [
                        "gpt-3.5-turbo-0125",
                        "gpt-4-0125-preview",
                        "claude-3-haiku-20240307",
                        "claude-3-sonnet-20240229",
                        "claude-3-opus-20240229",
                        "databricks/dbrx-instruct:nitro",
                        "nousresearch/nous-hermes-2-mixtral-8x7b-sft",
                        # "cohere/command-r-plus",
                    ],
                    label="Model Name",
                    info="Choose the model to use",
                )
            with gr.Row():
                openai_api_key = gr.Textbox(
                    label="OpenAI API Key",
                    placeholder="Enter your OpenAI API key here",
                    type="password",
                )
                anthropic_api_key = gr.Textbox(
                    label="Anthropic API Key",
                    placeholder="Enter your Anthropic API key here",
                    type="password",
                )
                openrouter_api_key = gr.Textbox(
                    label="OpenRouter API Key",
                    placeholder="Enter your OpenRouter API key here",
                    type="password",
                )
            with gr.Row():
                num_questions = gr.Slider(
                    minimum=1,
                    maximum=340,
                    value=10,
                    label="Number of questions to run the benchmark on",
                )
            with gr.Row():
                run_button = gr.Button("Run Benchmark")
            with gr.Row():
                with gr.Accordion("Results", open=True):
                    result = gr.Dataframe()
            with gr.Row():
                with gr.Accordion("Summary", open=False):
                    summary = gr.Dataframe()

            run_button.click(
                run_benchmark_gradio,
                inputs=[
                    tool_name,
                    model_name,
                    num_questions,
                    openai_api_key,
                    anthropic_api_key,
                    openrouter_api_key,
                ],
                outputs=[result, summary],
            )


demo.queue(default_concurrency_limit=40).launch()