llm-graph-kit 0.1.0__tar.gz

llm_graph_kit-0.1.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.3
+Name: llm-graph-kit
+Version: 0.1.0
+Summary: Add your description here
+Requires-Dist: mlx-augllm>=1.5
+Requires-Python: >=3.13.9
+Project-URL: Homepage, https://github.com/ToPo-ToPo-ToPo/llm_graph
+Description-Content-Type: text/markdown
+
+# llm_graph
+LLMベースのエージェント作成ライブラリ

llm_graph_kit-0.1.0/README.md

@@ -0,0 +1,2 @@
+# llm_graph
+LLMベースのエージェント作成ライブラリ

llm_graph_kit-0.1.0/pyproject.toml

@@ -0,0 +1,22 @@
+[project]
+name = "llm-graph-kit"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.13.9"
+dependencies = [
+    "mlx-augllm>=1.5",
+]
+
+[project.urls]
+Homepage = "https://github.com/ToPo-ToPo-ToPo/llm_graph"
+
+[build-system]
+requires = ["uv_build >= 0.9.21"]
+build-backend = "uv_build"
+
+[dependency-groups]
+dev = [
+    "build>=1.4.0",
+    "twine>=6.2.0",
+]

llm_graph_kit-0.1.0/src/llm_graph_kit/__init__.py

@@ -0,0 +1,9 @@
+
+from .llm_graph_kit import LLMGraphKit, NodeState
+from .graph_logger import GraphLogger
+
+__all__ = [
+    "LLMGraphKit",
+    "NodeState",
+    "GraphLogger"
+]

llm_graph_kit-0.1.0/src/llm_graph_kit/graph_logger.py

@@ -0,0 +1,118 @@
+import textwrap
+from typing import Any
+
+class GraphLogger:
+    """
+    プログラムの内容に依存せず、ターミナル表示の「スタイル」を提供する汎用ロガー。
+    """
+
+    # 色設定
+    COLORS = {
+        "HEADER": '\033[95m', "BLUE": '\033[94m', "CYAN": '\033[96m',
+        "GREEN": '\033[92m', "YELLOW": '\033[93m', "RED": '\033[91m',
+        "ENDC": '\033[0m', "BOLD": '\033[1m'
+    }
+
+    @classmethod
+    def print_phase_header(cls, title: str, emoji: str = "🚀"):
+        """メインフェーズの開始を目立つように表示します"""
+        c = cls.COLORS
+        print(f"\n{c['BLUE']}{c['BOLD']}" + "="*70 + f"{c['ENDC']}")
+        print(f"{c['BLUE']}{c['BOLD']} {emoji}  {title} {c['ENDC']}")
+        print(f"{c['BLUE']}{c['BOLD']}" + "="*70 + f"{c['ENDC']}")
+
+    @classmethod
+    def print_subtask_start(cls, index: int, task_name: str):
+        """サブタスクの開始を表示します"""
+        c = cls.COLORS
+        print(f"\n{c['YELLOW']}┌── 🔸 Subtask {index} ──────────────────────────────────────────────────{c['ENDC']}")
+        print(f"{c['YELLOW']}│ Task: {c['ENDC']}{task_name}")
+        print(f"{c['YELLOW']}└──────────────────────────────────────────────────────────────────{c['ENDC']}")
+
+    @classmethod
+    def log(cls, style: str, content: Any, title: str = ""):
+        """
+        スタイルを指定してログを出力します。
+        
+        Args:
+            style (str): 表示スタイル ("header", "box", "list", "info", "success", "error", "code")
+            content (Any): 表示内容（文字列、リスト、辞書など）
+            title (str): タイトルやラベル（任意）
+        """
+        c = cls.COLORS
+        style = style.lower()
+        
+        # ---------------------------------------------------------
+        # 1. 
+        # ---------------------------------------------------------
+        if style == "response":
+            # タイトルが指定されていなければデフォルトを設定
+            display_title = title if title else "Generated Response"
+            
+            # 色設定 (ここでは CYAN を使用。GREEN にしたい場合は c['GREEN'] に変更可)
+            color = c['GREEN'] 
+            
+            print(f"\n{color}{c['BOLD']}🤖 {display_title}{c['ENDC']}")
+            print(f"{color}──────────────────────────────────────────────────────────────{c['ENDC']}")
+            
+            # 本文も色付きで表示
+            print(f"{color}{content}{c['ENDC']}")
+            
+            print(f"{color}──────────────────────────────────────────────────────────────{c['ENDC']}\n")
+
+        # ---------------------------------------------------------
+        # 3. list: 計画や手順の箇条書き
+        # ---------------------------------------------------------
+        elif style == "list":
+            if title:
+                print(f"\n{c['BOLD']}📋 {title}:{c['ENDC']}")
+            
+            if isinstance(content, list):
+                for i, item in enumerate(content, 1):
+                    print(f"{i}. {item}")
+            else:
+                print(f"- {content}")
+
+        # ---------------------------------------------------------
+        # 4. info: 一般的な情報、ツール選択など（1行表示推奨）
+        # ---------------------------------------------------------
+        elif style == "info":
+            # 辞書が渡された場合は Key: Value 形式で見やすく
+            if isinstance(content, dict):
+                print(f"{c['CYAN']}🛠  {title}{c['ENDC']}")
+                for k, v in content.items():
+                    print(f"Running {k}: {v}")
+            else:
+                label = f"{title}: " if title else ""
+                print(f"{c['CYAN']}ℹ️  {label}{c['BOLD']}{content}{c['ENDC']}")
+
+        # ---------------------------------------------------------
+        # 5. code / preview: 実行結果などの長文プレビュー
+        # ---------------------------------------------------------
+        elif style == "code" or style == "preview":
+            text = str(content)
+            # 長すぎる場合は省略表示
+            preview = textwrap.shorten(text, width=200, placeholder="...")
+            lines = preview.split('\n')
+            if len(lines) > 5:
+                preview = "\n".join(lines[:5]) + "\n... (more lines) ..."
+            
+            label = title if title else "Output"
+            print(f"{c['GREEN']}📄 {label}:\n{preview}{c['ENDC']}")
+            print(f"{c['GREEN']}──────────────────────────────────────{c['ENDC']}")
+
+        # ---------------------------------------------------------
+        # 6. success / error: 評価や完了通知
+        # ---------------------------------------------------------
+        elif style == "success":
+            print(f"{c['GREEN']}✅ {title}: {content}{c['ENDC']}")
+            
+        elif style == "error":
+            print(f"{c['RED']}❌ {title}: {content}{c['ENDC']}")
+
+        # ---------------------------------------------------------
+        # 7. fallback: 想定外のスタイル
+        # ---------------------------------------------------------
+        else:
+            prefix = f"[{title}] " if title else ""
+            print(f"{prefix}{content}")

llm_graph_kit-0.1.0/src/llm_graph_kit/llm_graph_kit.py

@@ -0,0 +1,196 @@
+from typing import Dict, Any, Callable, Union, Tuple
+
+# ステートの型定義
+NodeState = Dict[str, Any]
+
+class LLMGraph:
+    """
+    ノードとエッジで構成されるステートマシンエンジン。
+    Router関数だけでなく、ステートの値を直接参照するルーティングに対応しました。
+    """
+    # 定数定義
+    START = "__START__"
+    END = "__END__"
+
+    def __init__(self):
+        # ノード名 -> 関数
+        self.nodes: Dict[str, Callable[[NodeState], NodeState]] = {}
+        # ノード名 -> 次のノード名 または (条件, マッピング辞書)
+        # 条件は Callable(関数) または str(ステートのキー)
+        self.edges: Dict[str, Union[str, Tuple[Union[Callable, str], Dict[str, str]]]] = {} 
+        self.entry_point: str = ""
+        self.subgraphs: Dict[str, 'LLMGraph'] = {}
+
+    def add_node(self, name: str, func: Callable[[NodeState], NodeState]):
+        """ノードを登録します"""
+        self.nodes[name] = func
+    
+    def add_node_with_subgraph(self, name: str, func: Callable[[NodeState], NodeState], subgraph: 'LLMGraph'):
+        """サブグラフ構造を持つノードを登録します"""
+        self.nodes[name] = func
+        self.subgraphs[name] = subgraph
+
+    def add_edge(self, from_node: str, to_node: str):
+        """固定ルートを定義します"""
+        if from_node == self.START:
+            self.entry_point = to_node
+            return
+        self.edges[from_node] = to_node
+
+    def add_conditional_edge(
+        self, 
+        from_node: str, 
+        condition: Union[Callable[[NodeState], str], str], 
+        path_map: Dict[str, str]
+    ):
+        """
+        条件分岐ルートを定義します。
+        
+        Args:
+            from_node: 分岐元のノード名
+            condition: 
+                - 関数: NodeStateを受け取りシグナル(文字列)を返す
+                - 文字列: シグナルが格納されているNodeStateのキー名
+            path_map: { "シグナル": "行き先のノード名" } の辞書
+        """
+        self.edges[from_node] = (condition, path_map)
+
+    def add_subgraph(self, node_name: str, subgraph: 'LLMGraph'):
+        """可視化用にサブグラフを登録します"""
+        self.subgraphs[node_name] = subgraph
+
+    def run(self, initial_state: NodeState) -> NodeState:
+        """グラフを実行します"""
+        if not self.entry_point:
+            raise ValueError("Entry point not set. Use add_edge(Graph.START, 'node_name').")
+
+        current_node_name = self.entry_point
+        state = initial_state.copy()
+
+        while current_node_name != self.END:
+            if current_node_name not in self.nodes:
+                raise ValueError(f"Node '{current_node_name}' is not defined!")
+
+            # ノード実行
+            node_func = self.nodes[current_node_name]
+            new_data = node_func(state)
+            if new_data:
+                state.update(new_data)
+
+            # --- 次の行き先を決定 ---
+            edge_data = self.edges.get(current_node_name)
+
+            if edge_data is None:
+                current_node_name = self.END
+            
+            # 条件付きエッジ (Func/Key, Map)
+            elif isinstance(edge_data, tuple):
+                condition, path_map = edge_data
+                
+                # ★改良点: 関数なら実行、文字列ならステートから取得
+                if callable(condition):
+                    signal = condition(state)
+                elif isinstance(condition, str):
+                    signal = state.get(condition)
+                    if signal is None:
+                         raise ValueError(f"NodeState key '{condition}' not found for routing from '{current_node_name}'")
+                else:
+                    raise ValueError("Invalid condition type in edge")
+                
+                # 文字列化（Enum対応）
+                signal_str = str(signal).split('.')[-1] if hasattr(signal, 'name') else str(signal)
+
+                # マッピング解決
+                next_dest = None
+                for key, val in path_map.items():
+                    key_str = str(key).split('.')[-1] if hasattr(key, 'name') else str(key)
+                    if key_str == signal_str:
+                        next_dest = val
+                        break
+                
+                if next_dest:
+                    print(f"[Router decision]: '{signal_str}' => Go to [{next_dest}]")
+                    current_node_name = next_dest
+                else:
+                    raise ValueError(f"Router returned '{signal}', but not found in map: {path_map}")
+            
+            # 固定エッジ
+            else:
+                current_node_name = edge_data
+
+        return state
+    
+    # ==========================================================================
+    # 可視化 (Mermaid) [改良版]
+    # ==========================================================================
+    def get_graph_mermaid(self) -> str:
+        lines = ["graph TD"]
+        
+        # スタイル定義
+        lines.append("  %% Styles")
+        lines.append("  classDef startClass fill:#f9f,stroke:#333,stroke-width:2px,rx:10,ry:10;")
+        lines.append("  classDef endClass fill:#f96,stroke:#333,stroke-width:2px,rx:10,ry:10;")
+        lines.append("  classDef nodeClass fill:#e1f5fe,stroke:#0277bd,stroke-width:2px,rx:5,ry:5;")
+        lines.append("  classDef routerClass fill:#fff9c4,stroke:#fbc02d,stroke-width:2px,stroke-dasharray: 5 5,rhombus;")
+        lines.append("  classDef subStartClass fill:#eee,stroke:#999,stroke-width:1px,rx:5,ry:5;")
+
+        def render_content(graph_obj, prefix="", is_subgraph=False):
+            # START / END
+            style = "subStartClass" if is_subgraph else "startClass"
+            end_style = "subStartClass" if is_subgraph else "endClass"
+            lines.append(f"    {prefix}{self.START}(START):::{style}")
+            lines.append(f"    {prefix}{self.END}(END):::{end_style}")
+
+            # ノード
+            for node in graph_obj.nodes:
+                node_id = f"{prefix}{node}"
+                lines.append(f"    {node_id}[{node}]:::nodeClass")
+
+            # Entry Point
+            if graph_obj.entry_point:
+                lines.append(f"    {prefix}{self.START} --> {prefix}{graph_obj.entry_point}")
+
+            # Edges
+            for from_node, edge_data in graph_obj.edges.items():
+                from_id = f"{prefix}{from_node}"
+                
+                if isinstance(edge_data, tuple):
+                    condition, path_map = edge_data
+                    router_id = f"{prefix}router_{from_node}"
+                    
+                    # ラベルの決定（関数名 or キー名）
+                    if callable(condition):
+                        label = f"{{{condition.__name__}}}"
+                    else:
+                        label = f"{{{condition}}}" # ステートのキー名を表示
+                    
+                    lines.append(f"    {from_id} --> {router_id}{label}:::routerClass")
+                    
+                    for signal, to_node in path_map.items():
+                        to_id = f"{prefix}{to_node}"
+                        s_label = str(signal).split('.')[-1]
+                        lines.append(f"    {router_id} -- {s_label} --> {to_id}")
+                else:
+                    to_id = f"{prefix}{edge_data}"
+                    lines.append(f"    {from_id} --> {to_id}")
+
+        if self.subgraphs:
+            # サブグラフ描画
+            for node_name, sub_graph in self.subgraphs.items():
+                cluster_id = f"Sub_{node_name}"
+                lines.append(f"  subgraph {cluster_id} [\"{node_name}\"]")
+                lines.append("    direction TD")
+                lines.append(f"    style {cluster_id} fill:#fffde7,stroke:#fbc02d,stroke-width:2px")
+                render_content(sub_graph, prefix=f"{node_name}_", is_subgraph=True)
+                lines.append("  end")
+
+            # メイングラフ描画
+            lines.append("  subgraph Main [Main Workflow]")
+            lines.append("    style Main fill:none,stroke:none")
+            render_content(self, prefix="", is_subgraph=False)
+            lines.append("  end")
+        
+        else:
+            render_content(self, prefix="", is_subgraph=False)
+        
+        return "\n".join(lines)