youngjin-langchain-tools 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

youngjin_langchain_tools/handlers/streamlit_langgraph_handler.py

@@ -8,8 +8,8 @@ LangGraph agent responses in Streamlit applications.
 Replaces the deprecated StreamlitCallbackHandler for LangGraph-based agents.
 """
 
-from typing import Any, Dict, List, Optional, Union, Generator
-from dataclasses import dataclass, field
+from typing import Any, Dict, List, Optional, Generator
+from dataclasses import dataclass
 import logging
 import re
 
@@ -22,38 +22,33 @@ logger = logging.getLogger(__name__)
 # ============================================================
 ERROR_PATTERNS = {
     # OpenAI errors
-    r"AuthenticationError.*API key": {
-        "title": "🔑 API Key 오류",
-        "message": "API 키가 설정되지 않았거나 유효하지 않습니다.",
+    r"AuthenticationError.*API key|openai.*api.*key|OPENAI_API_KEY": {
+        "title": "🔑 OpenAI API Key 오류",
+        "message": "OpenAI API 키가 설정되지 않았거나 유효하지 않습니다.",
         "solution": [
-            "1. `.env` 파일에 `OPENAI_API_KEY=sk-...` 형태로 키를 설정하세요.",
-            "2. 또는 코드 상단에 직접 API 키를 입력하세요.",
-            "3. API 키는 https://platform.openai.com/api-keys 에서 발급받을 수 있습니다.",
+            "환경변수 `OPENAI_API_KEY`를 설정하거나 클라이언트 초기화 시 `api_key` 파라미터를 전달하세요.",
+            "API 키 발급: https://platform.openai.com/api-keys",
         ],
     },
     r"RateLimitError|rate_limit|429": {
         "title": "⏱️ Rate Limit 초과",
         "message": "API 요청 한도를 초과했습니다.",
         "solution": [
-            "1. 잠시 후 다시 시도해주세요.",
-            "2. API 사용량을 확인하세요: https://platform.openai.com/usage",
-            "3. 필요시 요금제를 업그레이드하세요.",
+            "잠시 후 다시 시도하거나, API 사용량 및 요금제를 확인하세요.",
         ],
     },
     r"InsufficientQuotaError|insufficient_quota|billing": {
         "title": "💳 크레딧 부족",
         "message": "API 크레딧이 부족합니다.",
         "solution": [
-            "1. 결제 정보를 확인하세요: https://platform.openai.com/account/billing",
-            "2. 크레딧을 충전하세요.",
+            "API 제공자의 결제 페이지에서 크레딧을 충전하세요.",
         ],
     },
     r"InvalidRequestError|invalid_request": {
         "title": "❌ 잘못된 요청",
         "message": "API 요청 형식이 올바르지 않습니다.",
         "solution": [
-            "1. 입력 데이터를 확인하세요.",
-            "2. 모델명이 올바른지 확인하세요.",
+            "입력 데이터와 모델명이 올바른지 확인하세요.",
         ],
     },
     # Anthropic errors
@@ -61,8 +56,8 @@ ERROR_PATTERNS = {
         "title": "🔑 Anthropic API Key 오류",
         "message": "Anthropic API 키가 설정되지 않았거나 유효하지 않습니다.",
         "solution": [
-            "1. `.env` 파일에 `ANTHROPIC_API_KEY=sk-ant-...` 형태로 키를 설정하세요.",
-            "2. API 키는 https://console.anthropic.com/ 에서 발급받을 수 있습니다.",
+            "환경변수 `ANTHROPIC_API_KEY`를 설정하세요.",
+            "API 키 발급: https://console.anthropic.com/",
         ],
     },
     # Google errors
@@ -70,8 +65,8 @@ ERROR_PATTERNS = {
         "title": "🔑 Google API Key 오류",
         "message": "Google API 키가 설정되지 않았거나 유효하지 않습니다.",
         "solution": [
-            "1. `.env` 파일에 `GOOGLE_API_KEY=...` 형태로 키를 설정하세요.",
-            "2. API 키는 https://aistudio.google.com/apikey 에서 발급받을 수 있습니다.",
+            "환경변수 `GOOGLE_API_KEY`를 설정하세요.",
+            "API 키 발급: https://aistudio.google.com/apikey",
         ],
     },
     # Network errors
@@ -79,18 +74,14 @@ ERROR_PATTERNS = {
         "title": "🌐 네트워크 오류",
         "message": "API 서버에 연결할 수 없습니다.",
         "solution": [
-            "1. 인터넷 연결을 확인하세요.",
-            "2. 방화벽/프록시 설정을 확인하세요.",
-            "3. API 서버 상태를 확인하세요.",
+            "인터넷 연결 및 방화벽/프록시 설정을 확인하세요.",
         ],
     },
     r"TimeoutError|timeout|timed out": {
         "title": "⏰ 시간 초과",
         "message": "API 요청이 시간 초과되었습니다.",
         "solution": [
-            "1. 네트워크 연결을 확인하세요.",
-            "2. 잠시 후 다시 시도해주세요.",
-            "3. 요청 크기를 줄여보세요.",
+            "네트워크 연결을 확인하고 잠시 후 다시 시도하세요.",
         ],
     },
     # Model errors
@@ -98,9 +89,7 @@ ERROR_PATTERNS = {
         "title": "🤖 모델 오류",
         "message": "지정된 모델을 찾을 수 없습니다.",
         "solution": [
-            "1. 모델명이 올바른지 확인하세요.",
-            "2. 해당 모델에 대한 접근 권한이 있는지 확인하세요.",
-            "3. 사용 가능한 모델 목록을 확인하세요.",
+            "모델명과 접근 권한을 확인하세요.",
         ],
     },
 }
@@ -140,6 +129,13 @@ class StreamlitLanggraphHandlerConfig:
     expand_new_thoughts: bool = True
     """Whether to expand the status container to show tool calls."""
 
+    collapse_completed_thoughts: bool = True
+    """Whether to automatically collapse completed thought containers."""
+
+    max_thought_containers: int = 4
+    """Maximum number of thought containers to show at once.
+    When exceeded, oldest thoughts are moved to a 'History' expander."""
+
     max_tool_content_length: int = 2000
     """Maximum length of tool output to display before truncating."""
 
@@ -213,6 +209,8 @@ class StreamlitLanggraphHandler:
         container: Any,
         *,
         expand_new_thoughts: bool = True,
+        collapse_completed_thoughts: bool = True,
+        max_thought_containers: int = 4,
         max_tool_content_length: int = 2000,
         show_tool_calls: bool = True,
         show_tool_results: bool = True,
@@ -228,6 +226,11 @@ class StreamlitLanggraphHandler:
                        Usually st.container() or similar.
             expand_new_thoughts: Whether to expand status container
                                  to show tool calls. Defaults to True.
+            collapse_completed_thoughts: Whether to collapse completed
+                                         thought containers. Defaults to True.
+            max_thought_containers: Maximum number of thought containers
+                                    to show at once. Oldest are moved to
+                                    'History' expander. Defaults to 4.
             max_tool_content_length: Maximum characters of tool output
                                      to display. Defaults to 2000.
             show_tool_calls: Whether to show tool call info. Defaults to True.
@@ -241,6 +244,8 @@ class StreamlitLanggraphHandler:
         else:
             self._config = StreamlitLanggraphHandlerConfig(
                 expand_new_thoughts=expand_new_thoughts,
+                collapse_completed_thoughts=collapse_completed_thoughts,
+                max_thought_containers=max_thought_containers,
                 max_tool_content_length=max_tool_content_length,
                 show_tool_calls=show_tool_calls,
                 show_tool_results=show_tool_results,
@@ -252,6 +257,9 @@ class StreamlitLanggraphHandler:
         self._final_response: str = ""
         self._status_container: Any = None
         self._response_placeholder: Any = None
+        self._history_placeholder: Any = None  # Placeholder for history expander
+        self._thought_history: List[Dict[str, Any]] = []  # History of old thoughts
+        self._current_thoughts: List[Dict[str, Any]] = []  # Current visible thoughts
 
     @property
     def config(self) -> StreamlitLanggraphHandlerConfig:
@@ -343,9 +351,13 @@ class StreamlitLanggraphHandler:
 
         # Reset state
         self._final_response = ""
+        self._thought_history = []
+        self._current_thoughts = []
 
         # Create UI components
         with self._container:
+            # History expander placeholder (will be shown when needed)
+            self._history_placeholder = st.empty()
             self._status_container = st.status(
                 self._config.thinking_label,
                 expanded=self._config.expand_new_thoughts
@@ -416,6 +428,50 @@ class StreamlitLanggraphHandler:
 
         yield {"type": "complete", "data": {"response": self._final_response}}
 
+    def _add_thought(self, thought_type: str, data: Dict[str, Any]) -> None:
+        """Add a thought and manage history if max_thought_containers exceeded."""
+        import streamlit as st
+
+        thought = {"type": thought_type, "data": data}
+        self._current_thoughts.append(thought)
+
+        # Check if we need to move old thoughts to history
+        max_containers = self._config.max_thought_containers
+        if len(self._current_thoughts) > max_containers:
+            # Move oldest thoughts to history
+            while len(self._current_thoughts) > max_containers:
+                old_thought = self._current_thoughts.pop(0)
+                self._thought_history.append(old_thought)
+
+            # Update history expander
+            self._update_history_expander()
+
+    def _update_history_expander(self) -> None:
+        """Update the history expander with old thoughts."""
+        import streamlit as st
+
+        if not self._thought_history:
+            return
+
+        with self._history_placeholder.container():
+            with st.expander(f"📜 History ({len(self._thought_history)} items)", expanded=False):
+                for thought in self._thought_history:
+                    if thought["type"] == "tool_call":
+                        st.markdown(
+                            f"{self._config.tool_call_emoji} "
+                            f"**{thought['data']['name']}**: `{thought['data']['args']}`"
+                        )
+                    elif thought["type"] == "tool_result":
+                        st.markdown(
+                            f"{self._config.tool_complete_emoji} "
+                            f"**{thought['data']['name']}** 완료"
+                        )
+                        content = thought['data']['content']
+                        if len(content) > 200:  # Shorter preview in history
+                            st.code(content[:200] + "\n... (truncated)", language="text")
+                        else:
+                            st.code(content, language="text")
+
     def _handle_updates(
         self,
         data: Dict[str, Any]
@@ -439,6 +495,9 @@ class StreamlitLanggraphHandler:
                             tool_name = tc.get('name', 'tool')
                             tool_args = tc.get('args', {})
 
+                            # Add to thought tracking
+                            self._add_thought("tool_call", {"name": tool_name, "args": tool_args})
+
                             with self._status_container:
                                 st.write(
                                     f"{self._config.tool_call_emoji} "
@@ -456,12 +515,17 @@ class StreamlitLanggraphHandler:
                         tool_name = msg.name
                         tool_content = str(msg.content) if hasattr(msg, 'content') else ""
 
+                        # Add to thought tracking
+                        self._add_thought("tool_result", {"name": tool_name, "content": tool_content})
+
                         with self._status_container:
                             st.write(
                                 f"{self._config.tool_complete_emoji} "
                                 f"**{tool_name}** 완료"
                             )
-                            with st.expander(f"📋 {tool_name} 결과 보기", expanded=False):
+                            # Collapse based on config
+                            expanded = not self._config.collapse_completed_thoughts
+                            with st.expander(f"📋 {tool_name} 결과 보기", expanded=expanded):
                                 if len(tool_content) > self._config.max_tool_content_length:
                                     st.code(
                                         tool_content[:self._config.max_tool_content_length]

{youngjin_langchain_tools-0.1.1.dist-info → youngjin_langchain_tools-0.1.3.dist-info}/METADATA

@@ -1,14 +1,14 @@
 Metadata-Version: 2.4
 Name: youngjin-langchain-tools
-Version: 0.1.1
+Version: 0.1.3
 Summary: LangGraph utilities for Streamlit - StreamlitLanggraphHandler and more
 Project-URL: Homepage, https://github.com/yourusername/youngjin-langchain-tools
 Project-URL: Documentation, https://github.com/yourusername/youngjin-langchain-tools#readme
 Project-URL: Repository, https://github.com/yourusername/youngjin-langchain-tools.git
 Project-URL: Issues, https://github.com/yourusername/youngjin-langchain-tools/issues
 Project-URL: Changelog, https://github.com/yourusername/youngjin-langchain-tools/releases
-Author-email: YoungJin <your-email@example.com>
-Maintainer-email: YoungJin <your-email@example.com>
+Author-email: CoCoRoF <gkfua00@gmail.com>
+Maintainer-email: CoCoRoF <gkfua00@gmail.com>
 License: Apache-2.0
 License-File: LICENSE
 Keywords: agents,ai,callback,handler,langchain,langgraph,llm,streamlit
@@ -16,12 +16,13 @@ Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.12
+Requires-Python: >=3.10
 Requires-Dist: langchain-core<1.2.7,>=1.0.0
 Requires-Dist: langchain<1.2.7,>=1.0.0
 Requires-Dist: pydantic>=2.0.0

{youngjin_langchain_tools-0.1.1.dist-info → youngjin_langchain_tools-0.1.3.dist-info}/RECORD

@@ -1,9 +1,9 @@
 youngjin_langchain_tools/__init__.py,sha256=S5GJtbYymhDuGtTibEG61Li9UvwvoFtnKm4mxjWNgnU,1310
 youngjin_langchain_tools/handlers/__init__.py,sha256=-vGk-m1fqOipJSe02ogcScE0K3pdVwO9A_EqBovPRxo,397
-youngjin_langchain_tools/handlers/streamlit_langgraph_handler.py,sha256=hd509HM1SjAhPRxXqZh1we_ynDRetXrrB9TFSENJWY8,18423
+youngjin_langchain_tools/handlers/streamlit_langgraph_handler.py,sha256=TDWjcgthh9eFPnufELbo4gzjUCCSYro6fUSRpwSXYuY,21540
 youngjin_langchain_tools/utils/__init__.py,sha256=LgSd7Gz2n5WgIhxaKHqpmQVUaBUDT_TTTw63hzoWGgM,272
 youngjin_langchain_tools/utils/config.py,sha256=hHvdxsn5ZFWRJvR6N7ODy94JbCkeJocsS58hmvxoK5I,1640
-youngjin_langchain_tools-0.1.1.dist-info/METADATA,sha256=3KyJIb5zWwslYNBXcZFRbyOT_CZP2q-0sx3DzMSAYoI,6920
-youngjin_langchain_tools-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-youngjin_langchain_tools-0.1.1.dist-info/licenses/LICENSE,sha256=fENUlkDDxJEn5c0u8mhVcz5ek-rTg2L4Kby1tMNchI8,11342
-youngjin_langchain_tools-0.1.1.dist-info/RECORD,,
+youngjin_langchain_tools-0.1.3.dist-info/METADATA,sha256=0hGrXQcNkFakDVdfvVdk14CXuciF7f5GR_RXP0QjM10,6959
+youngjin_langchain_tools-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+youngjin_langchain_tools-0.1.3.dist-info/licenses/LICENSE,sha256=fENUlkDDxJEn5c0u8mhVcz5ek-rTg2L4Kby1tMNchI8,11342
+youngjin_langchain_tools-0.1.3.dist-info/RECORD,,