airflow-chat 0.1.0a2__tar.gz → 0.1.0a4__tar.gz

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: airflow-chat
-Version: 0.1.0a2
+Version: 0.1.0a4
 Summary: An Apache Airflow plugin that enables AI-powered chat interactions with your Airflow instance through MCP integration and an intuitive UI.
 License: MIT
 Author: Hipposys

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/airflow_chat/plugins/airflow_chat.py

@@ -251,10 +251,10 @@ class LLMChatAgent:
             
         except requests.RequestException as e:
             print(f"Failed to initialize chat session: {e}")
-            return False, None
+            return False, e
         except Exception as e:
             print(f"Unexpected error initializing session: {e}")
-            return False, None
+            return False, e
     
     def stream_chat_response(self, message, conversation_id=None):
         """Stream response from FastAPI backend"""
@@ -372,8 +372,9 @@ class AirflowChatView(AppBuilderBaseView):
         # Return streaming response
         def generate():
             try:
-                if os.environ.get('INTERNAL_AI_ASSISTANT_SERVER', True):
-                    from app.server.llm import get_stream_agent_responce
+                if str(os.environ.get('INTERNAL_AI_ASSISTANT_SERVER', True))\
+                   .lower() == 'true':
+                    from .app.server.llm import get_stream_agent_responce
                     md_uri = str(settings.Session().bind.url).replace('postgresql+psycopg2', 'postgres')
                     loop = asyncio.new_event_loop()
                     asyncio.set_event_loop(loop)
@@ -443,10 +444,11 @@ class AirflowChatView(AppBuilderBaseView):
         conversation_id = str(uuid.uuid4())
         
         try:
-            if not os.environ.get('INTERNAL_AI_ASSISTANT_SERVER', True):
+            if not str(os.environ.get('INTERNAL_AI_ASSISTANT_SERVER', True))\
+               .lower() == 'true':
                 success, cookies = self.llm_agent.initialize_chat_session(conversation_id)
             else:
-                from app.databases.postgres import Database
+                from .app.databases.postgres import Database
                 md_uri = str(settings.Session().bind.url).replace('postgresql+psycopg2', 'postgres')
                 asyncio.run(Database.setup(md_uri))
                 print('Database setup complete')
@@ -460,7 +462,7 @@ class AirflowChatView(AppBuilderBaseView):
                 })
             else:
                 return jsonify({
-                    "error": "Failed to initialize chat session"
+                    "error": f"Failed to initialize chat session: {cookies}"
                 }), 500
                 
         except Exception as e:

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/airflow_chat/plugins/app/databases/postgres.py

@@ -2,7 +2,7 @@ import os
 
 from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
 
-from app.utils.singleton import Singleton
+from ..utils.singleton import Singleton
 
 
 class Database(metaclass=Singleton):

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/airflow_chat/plugins/app/models/__init__.py

@@ -2,13 +2,13 @@ import os
 model_type, model_id = os.environ.get('LLM_MODEL_ID', 'none:none').split(':', 1)
 
 if model_type == 'bedrock':
-    from app.models.inference.bedrock_model import ChatBedrock
+    from .inference.bedrock_model import ChatBedrock
     ChatModel = ChatBedrock
 elif model_type == 'antropic':
-    from app.models.inference.antropic_model import ChatAnthropic
+    from .inference.antropic_model import ChatAnthropic
     ChatModel = ChatAnthropic
 elif model_type == 'openai':
-    from app.models.inference.openai_model import ChatOpenAI
+    from .inference.openai_model import ChatOpenAI
     ChatModel = ChatOpenAI
 else:
     ChatModel = None

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/airflow_chat/plugins/app/server/chat.py

@@ -3,7 +3,7 @@ import uuid
 from fastapi import APIRouter, Request
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
-from app.server.llm import get_stream_agent_responce
+from .llm import get_stream_agent_responce
 
 
 chat_router = APIRouter()

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/airflow_chat/plugins/app/server/llm.py

@@ -6,9 +6,9 @@ from langgraph.prebuilt import create_react_agent
 from langchain_core.messages import HumanMessage, BaseMessage, \
     SystemMessage, ToolMessage, AIMessage, AIMessageChunk
 
-from app.databases.postgres import Database
-from app.models import ChatModel
-from app.utils.logger import Logger
+from ..databases.postgres import Database
+from ..models import ChatModel
+from ..utils.logger import Logger
 
 import os
 

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/airflow_chat/plugins/app/server/main.py

@@ -5,10 +5,10 @@ from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
 from starlette.middleware.sessions import SessionMiddleware
 
-from app.server.chat import chat_router
-from app.databases.postgres import Database
-from app.utils.config import Config
-from app.utils.logger import Logger
+from .chat import chat_router
+from ..databases.postgres import Database
+from ..utils.config import Config
+from ..utils.logger import Logger
 
 
 @asynccontextmanager

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/airflow_chat/plugins/app/utils/config.py

@@ -1,6 +1,6 @@
 import os
 
-from app.utils.singleton import Singleton
+from ..utils.singleton import Singleton
 
 
 class Config(metaclass=Singleton):

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/airflow_chat/plugins/app/utils/logger.py

@@ -1,7 +1,7 @@
 import sys
 import logging
 
-from app.utils.singleton import Singleton
+from ..utils.singleton import Singleton
 
 
 class Logger(metaclass=Singleton):

{airflow_chat-0.1.0a2 → airflow_chat-0.1.0a4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "airflow-chat"
-version = "0.1.0a2"
+version = "0.1.0a4"
 description = "An Apache Airflow plugin that enables AI-powered chat interactions with your Airflow instance through MCP integration and an intuitive UI."
 license = "MIT"
 authors = ["Hipposys"]