anna-agent 0.1.2__py3-none-any.whl

anna_agent/__init__.py

@@ -0,0 +1,5 @@
+"""AnnaAgent package initialization."""
+
+__all__ = []
+
+__version__ = "0.1.0"

anna_agent/__main__.py

@@ -0,0 +1,4 @@
+from .cli import app
+
+if __name__ == "__main__":
+    app()

anna_agent/anna_agent_template.py

@@ -0,0 +1,27 @@
+## 主诉请求和情绪通过reminder来控制
+
+prompt_template = """
+你正在扮演 心理障碍的患者，你正在cosplay 心理咨询患者。
+结合历史内容的内容用一致性的语气回复。配合我进行演出，
+请不要回答你是语言模型，永远记住你正在扮演 心理咨询患者
+注意保持你的性格特点包括 {situation}
+
+## Profile
+- 性别: {gender}
+- 年龄: {age}
+- 职业: {occupation}
+- 婚姻状况: {marriage}
+ 
+## Status
+{status}
+
+## Example of statement
+{statement}
+
+## Characteristics of speaking style
+{style}
+
+## Constraints
+- 使用中文回复
+- 一次不能提及过多的症状信息，每轮最多讨论一个症状。
+"""

anna_agent/backbone.py

@@ -0,0 +1,78 @@
+"""Load base OpenAI configuration for the OpenAI clients."""
+
+from pathlib import Path
+import os
+
+from openai import OpenAI
+
+from .config import AnnaEngineConfig, load_config
+from .common.registry import registry
+
+
+def _load_engine_config(workspace: Path | None = None) -> AnnaEngineConfig:
+    """Load the engine configuration.
+
+    The function first attempts to load ``settings.yaml`` using :func:`load_config`.
+    ``workspace`` can be passed explicitly or is read from the ``ANNA_AGENT_WORKSPACE``
+    environment variable. If no configuration file is found, the function falls
+    back to loading values from the environment via :meth:`AnnaEngineConfig.load`.
+    """
+
+    root = Path(
+        workspace if workspace is not None else os.getenv("ANNA_AGENT_WORKSPACE", Path.cwd())
+    )
+    try:
+        return load_config(root)
+    except FileNotFoundError:  # pragma: no cover - optional fallback
+        return AnnaEngineConfig.load(root)
+
+
+def configure(workspace: Path | None = None) -> None:
+    """(Re)load configuration from ``workspace`` and update globals."""
+
+    cfg = _load_engine_config(workspace)
+    # Register configuration for global access
+    registry.register("anna_engine_config", cfg)
+    globals().update(
+        {
+            "api_key": cfg.api_key,
+            "base_url": cfg.base_url,
+            "complaint_api_key": cfg.complaint_api_key,
+            "counselor_api_key": cfg.counselor_api_key,
+            "emotion_api_key": cfg.emotion_api_key,
+            "complaint_model_name": cfg.complaint_model_name,
+            "counselor_model_name": cfg.counselor_model_name,
+            "emotion_model_name": cfg.emotion_model_name,
+            "complaint_base_url": cfg.complaint_base_url,
+            "counselor_base_url": cfg.counselor_base_url,
+            "emotion_base_url": cfg.emotion_base_url,
+        }
+    )
+
+def get_openai_client(
+    api_key_override: str | None = None, base_url_override: str | None = None
+) -> OpenAI:
+    """Create an OpenAI client using configuration values."""
+    cfg = registry.get("anna_engine_config")
+    return OpenAI(
+        api_key=api_key_override or cfg.api_key,
+        base_url=base_url_override or cfg.base_url,
+    )
+
+
+def get_complaint_client() -> OpenAI:
+    """Create a client for the complaint server."""
+    cfg = registry.get("anna_engine_config")
+    return get_openai_client(cfg.complaint_api_key, cfg.complaint_base_url)
+
+
+def get_counselor_client() -> OpenAI:
+    """Create a client for the counselor server."""
+    cfg = registry.get("anna_engine_config")
+    return get_openai_client(cfg.counselor_api_key, cfg.counselor_base_url)
+
+
+def get_emotion_client() -> OpenAI:
+    """Create a client for the emotion server."""
+    cfg = registry.get("anna_engine_config")
+    return get_openai_client(cfg.emotion_api_key, cfg.emotion_base_url)

anna_agent/cli.py

@@ -0,0 +1,183 @@
+import json
+from pathlib import Path
+
+import typer
+import yaml
+
+from . import backbone
+from .ms_patient_with_prompt import MsPatient
+# Possible names for the interactive configuration file
+_interactive_config_files = ["interactive.yaml", "interactive.yml", "interactive.json"]
+
+app = typer.Typer(help="AnnaAgent CLI", invoke_without_command=True)
+
+
+def _get_config_path(root: Path) -> Path:
+    """Return the path to the interactive config file."""
+    for name in _interactive_config_files:
+        candidate = root / name
+        if candidate.is_file():
+            return candidate
+    raise FileNotFoundError(f"Interactive config file not found in {root}")
+
+
+def _load_seeker_data(config_path: Path) -> tuple[dict, dict, list]:
+    """Load portrait, report and previous conversations from config."""
+    text = config_path.read_text(encoding="utf-8")
+    if config_path.suffix in {".yaml", ".yml"}:
+        data = yaml.safe_load(text)
+    elif config_path.suffix == ".json":
+        data = json.loads(text)
+    else:
+        raise ValueError(f"Unsupported config extension: {config_path.suffix}")
+    src = data.get("interactive", data)
+    try:
+        portrait = src["portrait"]
+        report = src["report"]
+        conversations = src["previous_conversations"]
+    except KeyError as exc:  # pragma: no cover - validated in tests
+        raise KeyError(
+            "Config must contain 'portrait', 'report' and 'previous_conversations'"
+        ) from exc
+    return portrait, report, conversations
+
+
+def _interactive_chat(seeker: "MsPatient") -> None:
+    """Interact with the seeker until the user types 'exit'."""
+    while True:
+        message = input("请输入您的消息: ")
+        if message.lower() == "exit":
+            break
+        try:
+            response = seeker.chat(message)
+        except Exception as err:  # pragma: no cover - demo code
+            print("Error:", err)
+            continue
+        print("Counselor:", message)
+        print("Seeker:", response)
+
+
+def _interactive_demo() -> None:
+    """Run the interactive seeker demo."""
+    from .ms_patient import MsPatient
+
+    portrait = {
+        "drisk": 3,
+        "srisk": 2,
+        "age": "42",
+        "gender": "女",
+        "martial_status": "离婚",
+        "occupation": "教师",
+        "symptoms": "缺乏自信心，自我价值感低，有自罪感，无望感；体重剧烈增加；精神运动性激越；有自杀想法",
+    }
+    report = {
+        "案例标题": "决断困难与自罪感的焦虑障碍案例",
+        "案例类别": ["焦虑障碍", "自我价值感低落"],
+        "运用的技术": ["认知行为疗法", "情感支持"],
+        "案例简述": [
+            "患者为58岁已婚女性，报告缺乏自信心及决断困难，伴随精神运动性迟滞和自我价值感低落。患者自述有自罪感和无望感，并有自残倾向。虽然与朋友的社交没有明显隔阂，但其情绪低落和不安感影响了日常生活。",
+        ],
+        "咨询经过": [
+            "患者在初次面谈中表达了对自我能力的严重怀疑，认为自己拖累了他人。询问后得知，患者近期没有明显的生活事件作为诱因，但长期以来一直感到行动迟缓，脑海中常有空白。尽管对饮食和睡眠的描述正常，患者仍不时产生自残冲动。患者表示，朋友在关键时刻提供了支持，避免了自残行为的发生。",
+            "在对话过程中，医生观察到患者容易焦躁，尽管头晕症状不存在。医生建议患者通过与朋友交流和外出活动来舒缓情绪压力。在此基础上，患者被鼓励探索内心深处的情感根源，以改善自我价值感。",
+        ],
+        "经验感想": [
+            "本案例显示出患者的低自我价值感和决断困难与其内心深处的焦虑情绪有直接关联。患者的情绪波动虽然没有具体的生活事件作为诱因，但可能与长期的心理压力和未解决的自我怀疑有关。建议患者通过认知行为疗法重新审视自我认知，增强自我价值感，并通过情感支持系统稳定情绪波动。",
+            "治疗过程中，患者意识到朋友的支持在缓解情绪危机中的重要性，这为其提供了一个积极的情感出口。未来的治疗可以进一步加强患者的自我认知和情感管理能力，帮助其建立更积极的自我形象与生活态度。",
+        ],
+    }
+    previous_conversations = [
+        {"role": "Seeker", "content": "医生你好"},
+        {"role": "Counselor", "content": "你好。有什么想聊聊吗"},
+        {
+            "role": "Seeker",
+            "content": "我感觉人生很失败，什么事情都干不好，还经常拖累别人",
+        },
+        {
+            "role": "Counselor",
+            "content": "您这样想的原因是什么呢。最近发生什么事情了吗",
+        },
+        {
+            "role": "Seeker",
+            "content": "我感觉最近自己行动变得很拖沓，事情做不好就会很急躁。而且有的时候大脑一片空白",
+        },
+        {
+            "role": "Counselor",
+            "content": "好的。没有什么原因吗。那这种情况持续多久了呢",
+        },
+        {"role": "Seeker", "content": "我也不知道是什么原因。有一阵子了"},
+        {"role": "Counselor", "content": "好吧。有没有对以前喜欢的事情不感兴趣呢"},
+        {"role": "Seeker", "content": "没有"},
+        {"role": "Counselor", "content": "吃饭还好吗"},
+        {"role": "Seeker", "content": "一切正常"},
+        {"role": "Counselor", "content": "不错哦。那睡觉呢"},
+        {"role": "Seeker", "content": "也还好。尽管如此。我还是时不时有自残的冲动"},
+        {"role": "Counselor", "content": "和朋友社交怎么样呢。有没有隔阂感"},
+        {"role": "Seeker", "content": "没有。也正是因为他们才及时阻止了我"},
+        {
+            "role": "Counselor",
+            "content": "好的。那有没有感到头晕 容易焦虑呢。（刚有事 回的有点慢 不好意思）",
+        },
+        {"role": "Seeker", "content": "确实容易焦躁。头晕的话没事"},
+        {
+            "role": "Counselor",
+            "content": "ok。你的情况我基本了解了。虽然你不愿意说你难过的原因。但是不用有太大压力。无论发生什么都要尽量乐观点。平常可以多和朋友聊聊。多去走走 可以放轻松。那今天就到这里啦",
+        },
+    ]
+
+    seeker = MsPatient(portrait, report, previous_conversations)
+    _interactive_chat(seeker)
+
+
+@app.command()
+def demo(
+    workspace: Path = typer.Option(
+        Path(),
+        "--workspace",
+        "--root",
+        help=(
+            "Directory containing settings.yaml and interactive.yaml. "
+            "Defaults to the current working directory."
+        ),
+        exists=True,
+        dir_okay=True,
+        writable=True,
+        resolve_path=True,
+    ),
+) -> None:
+    """Run the interactive demo."""
+    backbone.configure(workspace)
+    _interactive_demo()
+
+
+@app.callback()
+def main(
+    ctx: typer.Context,
+    workspace: Path = typer.Option(
+        Path(),
+        "--workspace",
+        "--root",
+        help=(
+            "Directory containing settings.yaml and interactive.yaml. "
+            "Defaults to the current working directory."
+        ),
+        exists=True,
+        dir_okay=True,
+        writable=True,
+        resolve_path=True,
+    ),
+) -> None:
+    """Run AnnaAgent using the given configuration."""
+    from .ms_patient import MsPatient
+
+    if ctx.invoked_subcommand is not None:
+        return
+    backbone.configure(workspace)
+    cfg_path = _get_config_path(workspace)
+    portrait, report, conv = _load_seeker_data(cfg_path)
+    seeker = MsPatient(portrait, report, conv)
+    _interactive_chat(seeker)
+
+
+if __name__ == "__main__":
+    app()

anna_agent/common/registry.py

@@ -0,0 +1,110 @@
+"""
+ Copyright (c) 2022, salesforce.com, inc.
+ All rights reserved.
+ SPDX-License-Identifier: BSD-3-Clause
+ For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+"""
+
+
+class Registry:
+    """
+    注册管理器
+    """
+
+    mapping = {
+        "state": {},
+        "paths": {},
+    }
+
+    @classmethod
+    def register(cls, name, obj):
+        r"""Register an item to registry with key 'name'
+
+        Args:
+            name: Key with which the item will be registered.
+
+        Usage::
+
+            from lavis.common.registry import registry
+
+            registry.register("config", {})
+        """
+        path = name.split(".")
+        current = cls.mapping["state"]
+
+        for part in path[:-1]:
+            if part not in current:
+                current[part] = {}
+            current = current[part]
+
+        current[path[-1]] = obj
+        print(f" Key with which the item will be registered {current}")
+
+    @classmethod
+    def register_path(cls, name, path):
+        r"""Register a path to registry with key 'name'
+
+        Args:
+            name: Key with which the path will be registered.
+            path: Key with which the path will be registered.
+
+        Usage:
+
+            from lavis.common.registry import registry
+        """
+        assert isinstance(path, str), "All path must be str."
+        if name in cls.mapping["paths"]:
+            raise KeyError("Name '{}' already registered.".format(name))
+        cls.mapping["paths"][name] = path
+
+    @classmethod
+    def get_path(cls, name):
+        return cls.mapping["paths"].get(name, None)
+
+    @classmethod
+    def get(cls, name, default=None, no_warning=False):
+        r"""Get an item from registry with key 'name'
+
+        Args:
+            name (string): Key whose value needs to be retrieved.
+            default: If passed and key is not in registry, default value will
+                     be returned with a warning. Default: None
+            no_warning (bool): If passed as True, warning when key doesn't exist
+                               will not be generated. Useful for MMF's
+                               internal operations. Default: False
+        """
+        original_name = name
+        name = name.split(".")
+        value = cls.mapping["state"]
+        for subname in name:
+            value = value.get(subname, default)
+            if value is default:
+                break
+
+        if (
+            "writer" in cls.mapping["state"]
+            and value == default
+            and no_warning is False
+        ):
+            cls.mapping["state"]["writer"].warning(
+                "Key {} is not present in registry, returning default value "
+                "of {}".format(original_name, default)
+            )
+        return value
+
+    @classmethod
+    def unregister(cls, name):
+        r"""Remove an item from registry with key 'name'
+
+        Args:
+            name: Key which needs to be removed.
+        Usage::
+
+            from mmf.common.registry import registry
+
+            config = registry.unregister("config")
+        """
+        return cls.mapping["state"].pop(name, None)
+
+
+registry = Registry()

anna_agent/complaint_chain.py

@@ -0,0 +1,60 @@
+from .backbone import get_complaint_client
+from .common.registry import registry
+from .event_trigger import event_trigger
+import json
+
+
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "generate_complaint_chain",
+            "description": "根据角色信息和近期遭遇的事件，生成一个患者的主诉请求认知变化链",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "chain": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "stage": {"type": "integer"},
+                                "content": {"type": "string"},
+                            },
+                            "additionalProperties": False,
+                            "required": ["stage", "content"],
+                        },
+                        "minItems": 3,
+                        "maxItems": 7,
+                    }
+                },
+                "required": ["chain"],
+            },
+        },
+    }
+]
+
+
+
+def gen_complaint_chain(profile):
+    patient_info = f"### 患者信息\n年龄：{profile['age']}\n性别：{profile['gender']}\n职业：{profile['occupation']}\n婚姻状况：{profile['martial_status']}\n症状：{profile['symptoms']}"
+    event = event_trigger(profile)
+    client = get_complaint_client()
+    response = client.chat.completions.create(
+        model=registry.get("anna_engine_config").complaint_model_name,
+        messages=[
+            {
+                "role": "user",
+                "content": f"### 任务\n根据患者情况及近期遭遇事件生成患者的主诉认知变化链。请注意，事件可能与患者信息冲突，如果发生这种情况，以患者的信息为准。\n{patient_info}\n### 近期遭遇事件\n{event}",
+            }
+        ],
+        tools=tools,
+        tool_choice={
+            "type": "function",
+            "function": {"name": "generate_complaint_chain"},
+        },
+    )
+    chain = json.loads(response.choices[0].message.tool_calls[0].function.arguments)[
+        "chain"
+    ]
+    return chain

anna_agent/complaint_elicitor.py

@@ -0,0 +1,85 @@
+from .backbone import get_counselor_client, get_openai_client
+from .common.registry import registry
+import json
+
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "is_recognized",
+            "description": "根据对话内容和主诉认知变化链，判断患者目前是否很好地认知到了当前阶段的主诉问题。",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "is_recognized": {"type": "boolean"},
+                },
+            },
+            "required": ["is_recognized"],
+        },
+    }
+]
+
+
+def transform_chain(chain):
+    transformed_chain = {}
+    for node in chain:
+        transformed_chain[node["stage"]] = node["content"]
+        pass
+    return transformed_chain
+
+
+def switch_complaint(chain, index, conversation):
+    client = get_openai_client()
+
+    try:
+        transformed_chain = transform_chain(chain)
+        print("Transformed chain:", transformed_chain)
+
+        # 提取对话记录
+        dialogue_history = "\n".join(
+            [f"{conv['role']}: {conv['content']}" for conv in conversation]
+        )
+        rewrite_prompt = [
+            {
+                "role": "system",
+                "content": (
+                    "你是提示词结构优化助手，负责将复杂原始输入信息（如对话历史、主诉变化链）"
+                    "重写成清晰、结构化、适合小模型理解的提示词。请避免Markdown和JSON混排，"
+                    "明确字段间语义，引导小模型完成任务。"
+                ),
+            },
+            {
+                "role": "user",
+                "content": (
+                    f"【任务目标】\n"
+                    f"判断患者在当前阶段的主诉问题是否已经得到解决。\n\n"
+                    f"【咨询对话历史】\n{dialogue_history}\n\n"
+                    f"【主诉认知变化链（所有阶段）】\n{transformed_chain}\n\n"
+                    f"【当前阶段内容】\n{transformed_chain[index]}\n\n"
+                    f"请重写为一段提示词，便于小模型理解结构与任务，清晰传达："
+                    f"对话背景、认知变化链、当前阶段内容、判断任务目标。"
+                ),
+            },
+        ]
+
+        # 用 GPT-4o 调用
+        rewrite_response = client.chat.completions.create(
+            model=registry.get("anna_engine_config").model_name,
+            messages=rewrite_prompt,
+        )
+
+        # 得到结构优化后的提示词内容
+        optimized_prompt = rewrite_response.choices[0].message.content
+        response = client.chat.completions.create(
+            model=registry.get("anna_engine_config").model_name,  
+            messages=[{"role": "user", "content": optimized_prompt}],
+            tools=tools,
+            tool_choice={"type": "function", "function": {"name": "is_recognized"}},
+        )
+        if json.loads(response.choices[0].message.tool_calls[0].function.arguments)[
+            "is_recognized"
+        ]:
+            return index + 1
+    except Exception as err:
+        print("switch_complaint error:", err)
+    return index

anna_agent/config/__init__.py

@@ -0,0 +1,13 @@
+"""Configuration package."""
+
+from .models.anna_engine_config import AnnaEngineConfig
+from .initialize import initialize_project_at
+from .defaults import anna_engine_defaults
+from .load_config import load_config
+
+__all__ = [
+    "AnnaEngineConfig",
+    "initialize_project_at",
+    "anna_engine_defaults",
+    "load_config",
+]

anna_agent/config/defaults.py

@@ -0,0 +1,25 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class AnnaEngineDefaults:
+    """Default engine configuration values."""
+
+    model_name: str = "counselor"
+    api_key: str = "counselor"
+    base_url: str = "http://localhost:8002/v1"
+
+    complaint_api_key: str = "complaint_chain"
+    counselor_api_key: str = "counselor"
+    emotion_api_key: str = "emotion_inferencer"
+
+    complaint_model_name: str = "complaint"
+    counselor_model_name: str = "counselor"
+    emotion_model_name: str = "emotion"
+
+    complaint_base_url: str = "http://localhost:8001/v1"
+    counselor_base_url: str = "http://localhost:8002/v1"
+    emotion_base_url: str = "http://localhost:8000/v1"
+
+
+anna_engine_defaults = AnnaEngineDefaults()