airflow-chat 0.1.0a3__py3-none-any.whl → 0.1.0a4__py3-none-any.whl

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/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/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/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/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/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/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/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.0a3.dist-info → airflow_chat-0.1.0a4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: airflow-chat
-Version: 0.1.0a3
+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.0a3.dist-info → airflow_chat-0.1.0a4.dist-info}/RECORD

@@ -1,26 +1,26 @@
 airflow_chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airflow_chat/plugins/README.md,sha256=Hwmn7ohaPBIoeww2xQ_aSeB6Fy_EgkHXA3LYFi8beIk,21
 airflow_chat/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-airflow_chat/plugins/airflow_chat.py,sha256=aGtciZJigmLn0rdG27027x8MpgzRrTfvwobMALm4eX8,19636
+airflow_chat/plugins/airflow_chat.py,sha256=WEKwQ2KSRtcEwbvK6DCIiB8Zw4W6COSRveJqCzpGHuA,19728
 airflow_chat/plugins/app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airflow_chat/plugins/app/databases/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-airflow_chat/plugins/app/databases/postgres.py,sha256=0UYnV8Z4F33CubrFwkzuXoOhPi11IvhIuu1L1WsZKW8,1287
-airflow_chat/plugins/app/models/__init__.py,sha256=KmN9_vswQd53wm7I38eg5ps0E23YRDtgh_Y25rYLMCM,482
+airflow_chat/plugins/app/databases/postgres.py,sha256=YfM7scHesKDJ5JS3qjMVOCMT96Nn8tdtNBy7lsGFpIM,1285
+airflow_chat/plugins/app/models/__init__.py,sha256=9mNNM5KkN-W6EN_Mo706CKeyTFkyucpYuv_RPOwizbg,452
 airflow_chat/plugins/app/models/inference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airflow_chat/plugins/app/models/inference/antropic_model.py,sha256=8kC7cXjzGiy77eDnetq7h9SEfUfIEid4T6_2Hd8E5Hc,509
 airflow_chat/plugins/app/models/inference/bedrock_model.py,sha256=8ebK4oNwqIqhEIBSkVqzNcfQpU2-RtW9KJ_Lxd7yYqA,595
 airflow_chat/plugins/app/models/inference/openai_model.py,sha256=8iNDgJRXBDP9HR9rMFwYKg-paUUQK2NBJw25ZY0EFWk,534
 airflow_chat/plugins/app/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-airflow_chat/plugins/app/server/chat.py,sha256=Fvgq_OiILqOApCIFqEH7HaXMrnxMoQ99wohz3Y2fK5M,1054
-airflow_chat/plugins/app/server/llm.py,sha256=QH2q8hvrEHf-EyPIcdCtwQO2EzlQx2fsLKLjAX2fNeE,8725
-airflow_chat/plugins/app/server/main.py,sha256=x3FRFWA8TB4ShD-XIo6FQLT-AW_e5H8MSq_Ncc3FkkE,1288
+airflow_chat/plugins/app/server/chat.py,sha256=t9Ft2XGyUDkPye-Tu84J03xlZWdZJFLHZS8epqJAleo,1044
+airflow_chat/plugins/app/server/llm.py,sha256=yz0mMOs5uGcHX0AjPi3fjP-C1CPAOfjnDBZMFBtYmsw,8719
+airflow_chat/plugins/app/server/main.py,sha256=BqNf74KdFs5SzHnF9PWhYe807yYfUmEUsodRNc33rOQ,1272
 airflow_chat/plugins/app/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-airflow_chat/plugins/app/utils/config.py,sha256=l0FfBlneeTdI69X9tGXFblMH8EQPVVrJdCoH9Ukxe7Y,315
-airflow_chat/plugins/app/utils/logger.py,sha256=EC_bekFeVJHVKwGABffDhXSrpHqgm8r00Q27WETuo90,1056
+airflow_chat/plugins/app/utils/config.py,sha256=lcUf2GMuXakDuWfyeDKglH5U4t7mQeHnsp3Uk0Es1iM,313
+airflow_chat/plugins/app/utils/logger.py,sha256=VWHwDOgHODs7RCpeHQcKAjTiwDFJHXt87ozvPcD6onQ,1054
 airflow_chat/plugins/app/utils/singleton.py,sha256=bQCsVc8XPpM6etUbbg2_bKKReCjLm0bFh6OgXmoBO0I,1789
 airflow_chat/plugins/templates/chat_interface.html,sha256=ZnMeIeaknXf7ZQI7PRV6fyxyzMsOMUz8KmMaX4k_mJo,18719
-airflow_chat-0.1.0a3.dist-info/LICENSE,sha256=XFYCNJc3ykWUpIIuB6uHwgnWKWUm3iez5vxgFF352as,1069
-airflow_chat-0.1.0a3.dist-info/METADATA,sha256=rFWE-EY0nQ-lM3pS3XVbiTKLZKbMoexXWembNCA9zao,1605
-airflow_chat-0.1.0a3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-airflow_chat-0.1.0a3.dist-info/entry_points.txt,sha256=kBsx-McWOPxyGQ7fQETHUlPt863-SkuW6rV-pl5aFn0,84
-airflow_chat-0.1.0a3.dist-info/RECORD,,
+airflow_chat-0.1.0a4.dist-info/LICENSE,sha256=XFYCNJc3ykWUpIIuB6uHwgnWKWUm3iez5vxgFF352as,1069
+airflow_chat-0.1.0a4.dist-info/METADATA,sha256=wJqB4QAO30QA8JKyLMusPnOEzJnu1UnY6uwAqgQT3zE,1605
+airflow_chat-0.1.0a4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+airflow_chat-0.1.0a4.dist-info/entry_points.txt,sha256=KzmeXDfhihgaLTQgh4vOz2ts7obTzZCg8CNgkHX1zaU,91
+airflow_chat-0.1.0a4.dist-info/RECORD,,

airflow_chat-0.1.0a4.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[airflow.plugins]
+airflow-chat-plugin=airflow_chat.plugins.airflow_chat:AirflowChatPlugin
+

airflow_chat-0.1.0a3.dist-info/entry_points.txt

@@ -1,3 +0,0 @@
-[airflow.plugins]
-airflow_chat=airflow_chat.plugins.airflow_chat:AirflowChatPlugin
-