mlx-augllm 1.5__tar.gz

mlx_augllm-1.5/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

mlx_augllm-1.5/PKG-INFO

@@ -0,0 +1,94 @@
+Metadata-Version: 2.4
+Name: mlx_augllm
+Version: 1.5
+Summary: ('A library for augmenting large language models using MLX',)
+Home-page: https://github.com/ToPo-ToPo-ToPo/mlx_augllm
+Author: Creator name
+License: Apache License Version 2.0
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: Apache Software License
+Requires-Python: >=3.13.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: mlx-lm>=0.30.0
+Requires-Dist: mlx-vlm>=0.3.9
+Requires-Dist: transformers>=5.0.0rc1
+Requires-Dist: torch>=2.9.1
+Requires-Dist: torchvision>=0.24.1
+Requires-Dist: pyyaml
+Requires-Dist: Pillow
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# mlx_augllm
+MLX（Apple Silicon向け機械学習フレームワーク）を用いた、
+ローカルで動作する LLM / VLM のための統一インターフェースライブラリです。
+
+本ライブラリは以下を目的としています。
+- ローカルLLM/VLMを簡単かつ一貫したAPIで扱える
+- Tool Use（関数呼び出し）に対応
+- 会話履歴の管理を自動化
+- Apple Siliconに最適化
+
+## インストール
+```
+pip install -U mlx_augllm
+```
+- Apple Silicon必須
+
+## サンプル
+```python
+from mlx_augllm import MlxAugmentedLLM, MlxLLMInterface, PromptBuilder 
+
+def run_test():
+
+    # モデルの準備
+    model_path = "mlx-community/gemma-3-27b-it-4bit"
+    augmented_llm = MlxAugmentedLLM(
+        llm_interface=MlxLLMInterface(
+            model_path=model_path,
+            use_vision=False,
+            temp=0.7,
+            top_k=50,
+            top_p=0.9,
+            min_p=0.05,
+            max_tokens=8192
+        ),
+        prompt_builder=PromptBuilder(system_prompt_text="あなたは有能なアシスタントです。"),
+    )
+
+    # 実行テスト
+    user_query = "トポロジー最適化について教えてください。"
+    
+    print(f"\nユーザーの問いかけ: {user_query}")
+    print("-" * 50)
+    print("AIの応答 (Streaming):")
+
+    # respond の呼び出し (contextを渡す)
+    response_generator = augmented_llm.respond(
+        user_text=user_query,
+        stream=True,
+        temp=0.7
+    )
+
+    full_response = ""
+    for chunk in response_generator:
+        print(chunk, end="", flush=True)
+        full_response += chunk
+    
+    print("\n" + "-" * 50)
+    print("【内部レポート】")
+    if augmented_llm.report_text:
+        print(f"最終回答の文字数: {len(augmented_llm.report_text)}")
+
+if __name__ == "__main__":
+    run_test()
+```

mlx_augllm-1.5/README.md

@@ -0,0 +1,64 @@
+# mlx_augllm
+MLX（Apple Silicon向け機械学習フレームワーク）を用いた、
+ローカルで動作する LLM / VLM のための統一インターフェースライブラリです。
+
+本ライブラリは以下を目的としています。
+- ローカルLLM/VLMを簡単かつ一貫したAPIで扱える
+- Tool Use（関数呼び出し）に対応
+- 会話履歴の管理を自動化
+- Apple Siliconに最適化
+
+## インストール
+```
+pip install -U mlx_augllm
+```
+- Apple Silicon必須
+
+## サンプル
+```python
+from mlx_augllm import MlxAugmentedLLM, MlxLLMInterface, PromptBuilder 
+
+def run_test():
+
+    # モデルの準備
+    model_path = "mlx-community/gemma-3-27b-it-4bit"
+    augmented_llm = MlxAugmentedLLM(
+        llm_interface=MlxLLMInterface(
+            model_path=model_path,
+            use_vision=False,
+            temp=0.7,
+            top_k=50,
+            top_p=0.9,
+            min_p=0.05,
+            max_tokens=8192
+        ),
+        prompt_builder=PromptBuilder(system_prompt_text="あなたは有能なアシスタントです。"),
+    )
+
+    # 実行テスト
+    user_query = "トポロジー最適化について教えてください。"
+    
+    print(f"\nユーザーの問いかけ: {user_query}")
+    print("-" * 50)
+    print("AIの応答 (Streaming):")
+
+    # respond の呼び出し (contextを渡す)
+    response_generator = augmented_llm.respond(
+        user_text=user_query,
+        stream=True,
+        temp=0.7
+    )
+
+    full_response = ""
+    for chunk in response_generator:
+        print(chunk, end="", flush=True)
+        full_response += chunk
+    
+    print("\n" + "-" * 50)
+    print("【内部レポート】")
+    if augmented_llm.report_text:
+        print(f"最終回答の文字数: {len(augmented_llm.report_text)}")
+
+if __name__ == "__main__":
+    run_test()
+```

mlx_augllm-1.5/mlx_augllm/__init__.py

@@ -0,0 +1,14 @@
+
+from .mlx_llm import MlxLLM
+from .mlx_augmented_llm import MlxAugmentedLLM, MlxLLMInterface, PromptBuilder, ChatMemory
+from .test_tools import CalculatorTool, UserGreetingTool
+
+__all__ = [
+    "MlxLLM",
+    "MlxAugmentedLLM",
+    "MlxLLMInterface",
+    "PromptBuilder",
+    "ChatMemory",
+    "CalculatorTool",
+    "UserGreetingTool"
+]

mlx_augllm-1.5/mlx_augllm/chat_memory.py

@@ -0,0 +1,73 @@
+import copy
+import pickle
+from typing import List, Dict, Tuple
+from mlx_vlm.utils import load_image
+
+#--------------------------------------------------------------------------------------------
+#
+#--------------------------------------------------------------------------------------------
+class ChatMemory:
+
+    def __init__(self, max_memory: int = 5):
+        self.max_memory = max_memory
+        self.history: List[Tuple[Dict, Dict]] = []
+
+    def add(self, user_text: str, assistant_text: str, user_image_paths: List[str] = None):
+        """新しい会話セットを追加"""
+        if self.max_memory <= 0:
+            return
+
+        user_msg = {"role": "user", "content": user_text}
+        if user_image_paths:
+            user_msg["images"] = user_image_paths
+
+        assistant_msg = {"role": "assistant", "content": assistant_text}
+        
+        self.history.append((user_msg, assistant_msg))
+
+        # 最大数を超えたら古いものを削除
+        if len(self.history) > self.max_memory:
+            self.history = self.history[-self.max_memory:]
+
+    def clear(self):
+        """履歴をリセット"""
+        self.history = []
+
+    def get_messages(self, use_vision: bool = False) -> Tuple[List[Dict], List]:
+        """
+        LLMに渡すためのメッセージリストと、対応するPIL画像のリストを生成
+        return: (flattened_messages, all_pils)
+        """
+        flattened = []
+        all_pils = []
+
+        for user_msg, assistant_msg in self.history:
+            # ユーザーメッセージの処理
+            u_msg = copy.deepcopy(user_msg)
+            if use_vision and "images" in u_msg:
+                paths = u_msg["images"]
+                pils = [load_image(str(p)) for p in paths]
+                
+                # VLM形式への変換
+                content = [{"type": "text", "text": u_msg["content"]}]
+                for _ in pils:
+                    content.append({"type": "image"})
+                
+                u_msg["content"] = content
+                all_pils.extend(pils)
+                del u_msg["images"]
+            
+            flattened.append(u_msg)
+            flattened.append(copy.deepcopy(assistant_msg))
+
+        return flattened, all_pils
+    
+    def save_to_file(self, filepath: str):
+        """現在の記憶をファイルに保存"""
+        with open(filepath, 'wb') as f:
+            pickle.dump(self.history, f)
+
+    def load_from_file(self, filepath: str):
+        """ファイルから記憶を復元"""
+        with open(filepath, 'rb') as f:
+            self.history = pickle.load(f)

mlx_augllm-1.5/mlx_augllm/function_calling.py

@@ -0,0 +1,212 @@
+
+
+# Author: Shun Ogawa (a.k.a. "ToPo")
+# Copyright (c) 2025 Shun Ogawa (a.k.a. "ToPo")
+# License: Apache License Version 2.0
+
+import inspect
+from typing import Callable, Any, Dict, List, Optional
+import yaml
+import re
+from .tool import Tool
+#------------------------------------------------------------------------------------
+# 関数 (ツール) を名前にマッピングして登録します。
+#------------------------------------------------------------------------------------
+def register_tools(tools: List[Tool]) -> Dict[str, Tool]:
+    return {tool.name(): tool for tool in tools}
+
+#------------------------------------------------------------------------------------
+# 呼び出し可能なターゲット (関数やメソッド) を取得します。
+# 関数、クラス、インスタンスを受け取って、signatureを取得可能な対象を返す
+#------------------------------------------------------------------------------------
+def get_callable_target(fn_or_instance: Any) -> Callable:
+    #
+    if inspect.isfunction(fn_or_instance):
+        return fn_or_instance
+    # ツールインスタンスの場合、_runメソッドを優先
+    elif hasattr(fn_or_instance, "_run") and callable(getattr(fn_or_instance, "_run")):
+        return getattr(fn_or_instance, "_run")
+    elif hasattr(fn_or_instance, "run") and callable(getattr(fn_or_instance, "run")):
+        return getattr(fn_or_instance, "run")
+    elif inspect.isclass(fn_or_instance):
+        if hasattr(fn_or_instance, "run") and callable(getattr(fn_or_instance, "run")): # クラスメソッド run
+            return getattr(fn_or_instance, "run")
+        # クラスの __call__ (通常は __init__) はツールの実行シグネチャとは異なるため、ここでは対象外とする
+        raise TypeError(f"クラス {fn_or_instance} から実行可能なターゲットメソッド (run または _run) を特定できませんでした。")
+    elif callable(fn_or_instance): # その他の callable オブジェクト
+        return fn_or_instance
+    else:
+        raise TypeError(f"{fn_or_instance} は関数、クラス、または適切な callable ではありません。")
+
+#------------------------------------------------------------------------------------
+# LLMにツール呼び出しのルールと関数一覧を伝えるシステムプロンプトを生成します。（YAML版）
+#------------------------------------------------------------------------------------
+def generate_system_prompt(functions: Dict[str, Callable]) -> str:
+    #
+    descriptions = []
+    for name, tool_instance in functions.items():
+        try:
+            # tool_instanceの_runメソッドのシグネチャを取得する
+            target_callable = get_callable_target(tool_instance)
+            sig = inspect.signature(target_callable)
+            
+            params_list = []
+            for k, v in sig.parameters.items():
+                if k == 'self': # 'self' パラメータは除外
+                    continue
+                param_type = v.annotation.__name__ if v.annotation != inspect.Parameter.empty else 'Any'
+                if v.default != inspect.Parameter.empty:
+                    params_list.append(f"{k}: {param_type} (optional, default={v.default})")
+                else:
+                    params_list.append(f"{k}: {param_type}")
+            params = ", ".join(params_list)
+            
+            # ツールのdescriptionを使用
+            docstring = tool_instance.description().strip()
+            descriptions.append(f"- `{name}({params})`: {docstring}")
+        except Exception as e:
+            descriptions.append(f"- `{name}()`: <定義情報の取得に失敗: {e}>")
+
+    # Function calling用のプロンプトを作成
+    prompt = f"""ユーザーの指示に応じて、以下のツールを呼び出します。
+### ツール呼び出しのルール（厳守）：
+
+1. ユーザーの指示に基づいてツールが必要な場合、以下の**厳密な YAML 形式**のみで返答してください：
+
+```yaml
+tool_calls:
+  - tool:
+      name: <ツール名>
+      arguments:
+        <引数名1>: <値1>
+        <引数名2>: <値2>
+        ...
+```
+- `tool_calls` はリストです。
+- 各要素は `tool` オブジェクトを含みます。
+- `tool` オブジェクトは `name` と `arguments` を含みます。
+- `arguments` は引数名をキー、値をバリューとするマップ（辞書）です。
+- YAMLコードブロック (```yaml ... ```) で囲んでください。
+
+ツール一覧：
+{chr(10).join(descriptions)}
+
+2. 関数を使う必要がないときは通常の自然言語で返答してください。
+
+この2つのルールを厳守して、ユーザーの要求に最適な形で応答してください。
+"""
+    return prompt
+
+#------------------------------------------------------------------------------------
+# 関数のシグネチャに基づいて、生の引数辞書を適切な型にキャストします。
+#------------------------------------------------------------------------------------
+def cast_arguments(func_callable: Callable, raw_args: dict) -> dict:
+    sig = inspect.signature(func_callable)
+    casted_args = {}
+
+    valid_param_names = {name for name in sig.parameters if name != 'self'}
+
+    for name, value in raw_args.items():
+        if name not in valid_param_names:
+            print(f"Warning: Argument '{name}' is not a valid parameter for {func_callable.__name__}. It will be ignored.")
+            continue
+
+        param = sig.parameters[name]
+        target_type = param.annotation
+
+        if target_type == inspect.Parameter.empty or target_type == Any:
+            casted_args[name] = value
+            continue
+
+        try:
+            if target_type == str:
+                casted_args[name] = str(value)
+            elif target_type == int:
+                casted_args[name] = int(value)
+            elif target_type == float:
+                casted_args[name] = float(value)
+            elif target_type == bool:
+                if isinstance(value, str):
+                    casted_args[name] = value.lower() in ['true', 'yes', 'on', '1']
+                else:
+                    casted_args[name] = bool(value)
+            else: # リスト、辞書などはYAMLパーサーが適切に変換していることを期待
+                casted_args[name] = value
+        except (ValueError, TypeError) as e:
+            print(f"Warning: Failed to cast argument '{name}' (value: '{value}') to type {target_type.__name__}. Using original value. Error: {e}")
+            casted_args[name] = value
+
+    # デフォルト値を持つ引数が省略された場合の処理
+    for name, param in sig.parameters.items():
+        if name == 'self':
+            continue
+        if name not in casted_args and param.default is not inspect.Parameter.empty:
+            casted_args[name] = param.default
+        elif name not in casted_args and param.default is inspect.Parameter.empty : #必須引数が指定されてない
+            print(f"Warning: Required argument '{name}' for {func_callable.__name__} was not provided by LLM.")
+            # ここでエラーをraiseするか、Noneを設定するかは設計による
+            # casted_args[name] = None # またはエラー
+
+    return casted_args
+
+#------------------------------------------------------------------------------------
+# LLMの応答からYAML形式のtool_callsセクションを抽出・解析します。
+#------------------------------------------------------------------------------------
+def extract_tool_calls_from_yaml(content: str) -> List[Dict[str, Any]]:
+    content = content.strip()
+
+    # YAMLコードブロックの検出
+    match = re.search(r"```yaml\s*(.*?)\s*```", content, re.DOTALL)
+
+    yaml_str: Optional[str]
+    if match:
+        yaml_str = match.group(1).strip()
+    elif content.startswith("tool_calls:"): # YAMLブロックなしで直接YAMLが始まる場合も許容
+        yaml_str = content
+    else:
+        yaml_str = None
+
+    if not yaml_str:
+        raise ValueError("YAMLコードブロックまたはtool_callsで始まるYAML文字列が見つかりませんでした。")
+
+    try:
+        parsed = yaml.safe_load(yaml_str)
+
+        if not isinstance(parsed, dict) or "tool_calls" not in parsed:
+            raise ValueError("YAMLのルートが辞書型でなく、'tool_calls'キーが含まれていません。")
+
+        calls = parsed["tool_calls"] # 修正: キー名からコロンを削除
+        if not isinstance(calls, list):
+            # LLMが単一のツールコールをリストでなく直接記述した場合を許容する（例：tool_calls: {tool: ...}）
+            if isinstance(calls, dict) and "tool" in calls:
+                calls = [calls] # リストに変換
+            else:
+                raise ValueError("'tool_calls'の値はリストである必要があります。")
+
+        extracted_tool_calls = []
+        for item in calls:
+            if not isinstance(item, dict) or "tool" not in item:
+                raise ValueError("各tool_callエントリは辞書型で、'tool'キーを含む必要があります。")
+
+            tool_data = item["tool"] # 修正: キー名からコロンを削除
+            if not isinstance(tool_data, dict):
+                raise ValueError("'tool'の値は辞書型である必要があります。")
+
+            name = tool_data.get("name") # 修正: .getを使用し、キー名からコロンを削除
+            arguments = tool_data.get("arguments", {}) # 修正: .getを使用し、引数がない場合は空の辞書をデフォルトに
+
+            if not isinstance(name, str) or not name: # 名前が文字列で、空でないことを確認
+                raise ValueError(f"無効なツール名です: '{name}'。ツール名は文字列である必要があります。")
+            if not isinstance(arguments, dict):
+                raise ValueError(f"ツールの引数 ('arguments') は辞書型である必要がありますが、'{type(arguments)}'でした (ツール: {name})。")
+
+            # OpenAI互換の形式に少し寄せる
+            extracted_tool_calls.append({
+                "function": {"name": name, "arguments": arguments} 
+            })
+        return extracted_tool_calls
+
+    except yaml.YAMLError as e:
+        raise ValueError(f"YAML解析エラー: {e}\n解析対象のYAML文字列:\n---\n{yaml_str}\n---")
+    except Exception as e: # その他のパース関連エラー
+        raise ValueError(f"tool_callsの解析中に予期せぬエラーが発生しました: {e}\n解析対象のYAML文字列:\n---\n{yaml_str}\n---")