asktable-advisor 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

asktable_advisor/__main__.py

@@ -1,29 +1,39 @@
 """Command-line interface for AskTable Advisor."""
 
-import sys
 import logging
-from pathlib import Path
+import sys
 
 from rich.console import Console
-from rich.panel import Panel
 from rich.markdown import Markdown
+from rich.panel import Panel
 from rich.prompt import Prompt
 
-from .config import AdvisorSettings, get_settings
 from .agent.advisor import AdvisorAgent
+from .config import get_settings
 
 # Setup console
 console = Console()
 
-# Setup logging
-logging.basicConfig(
-    level=logging.INFO,
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-    handlers=[
-        logging.FileHandler("asktable_advisor.log"),
-        logging.StreamHandler(sys.stderr),
-    ],
-)
+
+def setup_logging(log_level: str = "WARNING") -> None:
+    """
+    Setup logging configuration.
+
+    Args:
+        log_level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
+    """
+    # Convert string to logging level
+    numeric_level = getattr(logging, log_level.upper(), logging.WARNING)
+
+    logging.basicConfig(
+        level=numeric_level,
+        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+        handlers=[
+            logging.FileHandler("asktable_advisor.log"),
+            logging.StreamHandler(sys.stderr),
+        ],
+        force=True,  # Override any existing configuration
+    )
 
 
 def print_welcome() -> None:
@@ -128,6 +138,9 @@ def main() -> None:
         # Load settings
         settings = get_settings()
 
+        # Setup logging with configured level
+        setup_logging(settings.log_level)
+
         # Print welcome
         print_welcome()
 

asktable_advisor/__version__.py

@@ -1,4 +1,4 @@
 """Version information for asktable-advisor."""
 
-__version__ = "1.0.0"
+__version__ = "1.0.3"
 __version_info__ = tuple(int(x) for x in __version__.split("."))

asktable_advisor/agent/llm_client.py

@@ -36,8 +36,9 @@ class LLMClient:
 
         # Initialize Anthropic client with Aliyun Bailian endpoint
         self.client = Anthropic(
-            api_key=settings.anthropic_api_key,
-            base_url=settings.anthropic_base_url,
+            api_key=settings.advisor_anthropic_api_key,
+            base_url=settings.advisor_anthropic_base_url,
+            timeout=60.0,  # Set 60 second timeout to avoid streaming requirement
         )
 
         self.model = settings.agent_model
@@ -46,7 +47,7 @@ class LLMClient:
 
         logger.info(
             f"LLM client initialized: model={self.model}, "
-            f"base_url={settings.anthropic_base_url}"
+            f"base_url={settings.advisor_anthropic_base_url}"
         )
 
     def create_message(
@@ -89,7 +90,8 @@ class LLMClient:
                 # TODO: Implement streaming response handling
                 logger.warning("Streaming not yet implemented, falling back to non-streaming")
 
-            response = self.client.messages.create(**params)
+            # Set explicit timeout to avoid streaming requirement for large max_tokens
+            response = self.client.messages.create(**params, timeout=60.0)
 
             logger.debug(
                 f"Received response: "

asktable_advisor/config.py

@@ -22,11 +22,11 @@ class AdvisorSettings(BaseSettings):
     )
 
     # LLM API Configuration (Aliyun Bailian - qwen3-max)
-    anthropic_base_url: str = Field(
+    advisor_anthropic_base_url: str = Field(
         default="https://dashscope.aliyuncs.com/apps/anthropic",
         description="Anthropic-compatible API base URL (Aliyun Bailian)",
     )
-    anthropic_api_key: str = Field(
+    advisor_anthropic_api_key: str = Field(
         ...,
         description="DashScope API key for Aliyun Bailian",
     )
@@ -45,6 +45,12 @@ class AdvisorSettings(BaseSettings):
         description="Maximum tokens for LLM responses",
     )
 
+    # Logging Configuration
+    log_level: str = Field(
+        default="WARNING",
+        description="Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)",
+    )
+
     # MySQL Database Configuration
     mysql_host: str = Field(..., description="MySQL host")
     mysql_port: int = Field(default=3306, description="MySQL port")

{asktable_advisor-1.0.1.dist-info → asktable_advisor-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: asktable-advisor
-Version: 1.0.1
+Version: 1.0.3
 Summary: Conversational AI Agent for building AskTable demo scenarios
 Author-email: DataMini <contact@asktable.com>
 Maintainer-email: DataMini <contact@asktable.com>

{asktable_advisor-1.0.1.dist-info → asktable_advisor-1.0.3.dist-info}/RECORD

@@ -1,10 +1,10 @@
 asktable_advisor/__init__.py,sha256=21OEclDkj7fT8swRC2d7JZAqB2cB52lt-LLJno5eVFY,442
-asktable_advisor/__main__.py,sha256=WIDnjg0dQL6WykpBCd40MTMX4qd74LNaau_RfoAAZ2w,4213
-asktable_advisor/__version__.py,sha256=2knl3SnRO4uZ42QxU1rk-tlvYWJ_geIulHlf53SOxEU,136
-asktable_advisor/config.py,sha256=x4SF5r3d2Ff2Wfaq7R_5thllPTyXtow65yB7qZpgry8,2484
+asktable_advisor/__main__.py,sha256=7hZeayoibCKUeGIZSgU8Pg9N-OejQJ35kq8A-yRBFT8,4637
+asktable_advisor/__version__.py,sha256=GGOC7DDZK2TBiEaMeXqhGjpUQM4IMAHqHGzEXYJBIxA,136
+asktable_advisor/config.py,sha256=l2vPCaUIPUeaJt-2NhlOFIUB9pVWOx7uD58qgAOO1Bk,2667
 asktable_advisor/agent/__init__.py,sha256=QU-dutlZwnjeJ1EMtdjTLeia3SloykUAatiuHuEeHq4,147
 asktable_advisor/agent/advisor.py,sha256=g6hJpGicTyaUauWWPI6bwGsz5tH_AeaDvL6IkcSER9E,11349
-asktable_advisor/agent/llm_client.py,sha256=8t9T2BkjxHHyOchZPxuVrKjSnBy7tvDmV8eV9zG-pz4,5573
+asktable_advisor/agent/llm_client.py,sha256=f8nVteCXiXDF0TWlYVKEhpUXuvAskuK_qbc-ipLTZR4,5780
 asktable_advisor/agent/prompts.py,sha256=eJSeRhivDKN3Qtpl_hedqJiFtsccLuFFfUcsVCzmtVA,5467
 asktable_advisor/agent/tools.py,sha256=YJv1IZQ-OIrFP4FghxGNIL12uLZ-U-cKZxuKWZEZUEo,12875
 asktable_advisor/asktable/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -18,9 +18,9 @@ asktable_advisor/database/manager.py,sha256=0dMh7s7Xs-9vpC92VgIpDJSIiu_YxJApvnmi
 asktable_advisor/database/schema_generator.py,sha256=_K8FlJ3FMtATmxzYO-4Fogace5Zsvh2p8dcGBBNkjPw,4535
 asktable_advisor/knowledge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 asktable_advisor/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-asktable_advisor-1.0.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-asktable_advisor-1.0.1.dist-info/METADATA,sha256=RpkESjEK6mPcOlLPgZUfjhwMYlHW_cWiF1gL6pWulYw,7443
-asktable_advisor-1.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-asktable_advisor-1.0.1.dist-info/entry_points.txt,sha256=m4MJ7R-05wjqqPf-awcMj9JqtwgiK2Du8UV_l6VNh6k,68
-asktable_advisor-1.0.1.dist-info/top_level.txt,sha256=cDBvLmuBSm43p7q7GEoZoIsB0q2al1NZbbDktRe42bQ,17
-asktable_advisor-1.0.1.dist-info/RECORD,,
+asktable_advisor-1.0.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+asktable_advisor-1.0.3.dist-info/METADATA,sha256=g2d9-t-zESjGNW16g_Lkb2uR6pspjfjqL1LlRgjLSQQ,7443
+asktable_advisor-1.0.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+asktable_advisor-1.0.3.dist-info/entry_points.txt,sha256=m4MJ7R-05wjqqPf-awcMj9JqtwgiK2Du8UV_l6VNh6k,68
+asktable_advisor-1.0.3.dist-info/top_level.txt,sha256=cDBvLmuBSm43p7q7GEoZoIsB0q2al1NZbbDktRe42bQ,17
+asktable_advisor-1.0.3.dist-info/RECORD,,