Shift key inputs

modules/knowledge_retrieval/component.py

@@ -13,8 +13,8 @@ def determine_and_execute(api_key, question: str, temperature: float):
 examples = [["""When is my grandmothers birthday?""", 0.6], ["What was my tax burden last year?", 0.6 ], ["What is the most recent magic the gathering card set released?", 0.6], ["What products are the most popular with my small business customers?", 0.6]]
 
 def create_knowledge_router_ui(cache_examples=False):  
-    api_key = gr.Textbox(label="You OpenAI API key", type="password")
     with gr.Row():
+        api_key = gr.Textbox(label="You OpenAI API key", type="password")
         question = gr.Textbox(label="Enter your question here:")
         temperature = gr.Slider(minimum=0, maximum=2, default=.7, label="Temperature")
     with gr.Column():

modules/reasoning/component.py

@@ -19,8 +19,8 @@ He puts the cup down in the garden, then walks to the garage.
 Where is the ball?""", 0.6], ["Given the task of building a house in the middle of a river, what are three strategies I could use to mitigate risk of flooding?", 0.6 ]]
 
 def create_reasoning_router_ui(cache_examples=False): 
-    api_key = gr.Textbox(label="You OpenAI API key", type="password")
     with gr.Row():
+        api_key = gr.Textbox(label="You OpenAI API key", type="password")
         question = gr.Textbox(label="Enter your question here:")
         temperature = gr.Slider(minimum=0, maximum=2, default=.7, label="Temperature")
     with gr.Column():
@@ -29,5 +29,5 @@ def create_reasoning_router_ui(cache_examples=False):
     
     generate_button = gr.Button(label="Generate")
     generate_button.click(determine_and_execute, outputs=[reasoning_strategy, reasoning], inputs=[api_key, question, temperature])
-    gr.Examples(examples=examples, fn=determine_and_execute, cache_examples=cache_examples, inputs=[api_key, question, temperature], outputs=[reasoning_strategy, reasoning])
+    gr.Examples(examples=examples, fn=determine_and_execute, cache_examples=cache_examples, inputs=[question, temperature], outputs=[reasoning_strategy, reasoning])