abhishekbhakat/airflow-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Models list

Browse Source
This commit is contained in:
abhishekbhakat
2025-02-24 09:47:28 +00:00
parent 2a14475efc
commit 0b50ed0b85
2 changed files with 233 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
plugins/airflow-wingman/plugin.py
View File

@@ -16,21 +16,63 @@ class WingmanView(AppBuilderBaseView):
route_base = "/wingman"
default_view = "chat"
AVAILABLE_MODELS = {
"anthropic": {
"name": "Anthropic",
"endpoint": "https://api.anthropic.com/v1/messages",
"models": [
{
"id": "claude-3.5-sonnet",
"name": "Claude 3.5 Sonnet",
"default": True,
"context_window": 200000,
"description": "Input $3/M tokens, Output $15/M tokens",
},
{
"id": "claude-3.5-haiku",
"name": "Claude 3.5 Haiku",
"default": False,
"context_window": 200000,
"description": "Input $0.80/M tokens, Output $4/M tokens",
},
],
},
"openrouter": {
"name": "OpenRouter",
"endpoint": "https://openrouter.ai/api/v1/chat/completions",
"models": [
{
"id": "anthropic/claude-3.5-sonnet",
"name": "Claude 3.5 Sonnet",
"default": False,
"context_window": 200000,
"description": "Input $3/M tokens, Output $15/M tokens",
},
{
"id": "anthropic/claude-3.5-haiku",
"name": "Claude 3.5 Haiku",
"default": False,
"context_window": 200000,
"description": "Input $0.80/M tokens, Output $4/M tokens",
},
],
},
}
@expose("/")
def chat(self):
"""
Chat interface for Airflow Wingman.
"""
return self.render_template(
"wingman_chat.html",
title="Airflow Wingman"
"wingman_chat.html", title="Airflow Wingman", models=self.AVAILABLE_MODELS
)
# Create AppBuilder View
v_appbuilder_view = WingmanView()
v_appbuilder_package = {
"name": "Airflow Wingman",
"name": "Wingman",
"category": "AI",
"view": v_appbuilder_view,
}