ai_agents / server.py
jpfearnworks's picture
Add knowledge domain router
03f1c64
raw
history blame
820 Bytes
from dotenv import load_dotenv, find_dotenv
import os
import gradio as gr
from modules.reasoning.component import create_reasoning_router_ui
from modules.knowledge_retrieval.component import create_knowledge_router_ui
load_dotenv(find_dotenv())
openai_api_key = os.getenv("OPENAI_API_KEY")
def create_interface():
title: str = "Prompt Strategy Demo"
description: str = "AI Agents Sandbox"
with gr.Blocks(analytics_enabled=False, capture_session=True, title=title, description=description) as interface:
with gr.Tab("Reasoning Router"):
create_reasoning_router_ui()
with gr.Tab("Knowledge Domains"):
create_knowledge_router_ui()
interface.queue()
interface.launch(server_name="0.0.0.0", server_port=7000)
if __name__ == "__main__":
create_interface()