small fixes

deepsense-ai · Sep 4, 2024 · 29b15a0 · 29b15a0
1 parent f12f834
commit 29b15a0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/examples/visualize_views_code.py b/examples/visualize_views_code.py
@@ -11,14 +11,17 @@
 from dbally.llms.litellm import LiteLLM
 
 
-async def main():
+async def main() -> None:
     await country_similarity.update()
+
     llm = LiteLLM(model_name="gpt-3.5-turbo")
     dbally.event_handlers = [CLIEventHandler(), BufferEventHandler()]
+
     collection = dbally.create_collection("candidates", llm)
     collection.add(CandidateView, lambda: CandidateView(engine))
     collection.add(SampleText2SQLViewCyphers, lambda: SampleText2SQLViewCyphers(create_freeform_memory_engine()))
-    gradio_interface = await create_gradio_interface(user_collection=collection)
+
+    gradio_interface = create_gradio_interface(collection)
     gradio_interface.launch()
 
 

diff --git a/src/dbally/gradio/interface.py b/src/dbally/gradio/interface.py
@@ -216,7 +216,7 @@ def create_interface(self) -> gr.Interface:
                             interactive=bool(views),
                             max_lines=1,
                         )
-                        query = gr.Textbox(
+                        question = gr.Textbox(
                             label="Question",
                             placeholder="Enter your question",
                             interactive=bool(views),
@@ -226,14 +226,14 @@ def create_interface(self) -> gr.Interface:
                             label="Use Natural Language Responder",
                             interactive=bool(views),
                         )
-                        query_button = gr.Button(
+                        ask_button = gr.Button(
                             value="Ask",
                             variant="primary",
                             interactive=bool(views),
                         )
                         clear_button = gr.ClearButton(
                             value="Reset",
-                            components=[query],
+                            components=[question],
                             interactive=bool(views),
                         )
 
@@ -341,10 +341,10 @@ def create_interface(self) -> gr.Interface:
                     view_preview_label,
                 ],
             )
-            query_button.click(
+            ask_button.click(
                 fn=self._ask_collection,
                 inputs=[
-                    query,
+                    question,
                     model_name,
                     api_key,
                     natural_language_response_checkbox,