pycityagent 2.0.0a13__py3-none-any.whl → 2.0.0a15__py3-none-any.whl

pycityagent/utils/survey_util.py

@@ -0,0 +1,53 @@
+def process_survey_for_llm(survey_dict: dict) -> str:
+    """
+    将问卷字典转换为LLM可以逐题处理的格式，使用英文提示
+    """
+    prompt = f"""Survey Title: {survey_dict['title']}
+Survey Description: {survey_dict['description']}
+
+Please answer each question in the following format:
+
+"""
+    
+    question_count = 1
+    for page in survey_dict['pages']:
+        for question in page['elements']:
+            prompt += f"Question {question_count}: {question['title']}\n"
+            
+            # 根据不同类型的问题生成不同的提示
+            if question['type'] == 'radiogroup':
+                prompt += "Options: " + ", ".join(question['choices']) + "\n"
+                prompt += "Please select ONE option\n"
+            
+            elif question['type'] == 'checkbox':
+                prompt += "Options: " + ", ".join(question['choices']) + "\n"
+                prompt += "You can select MULTIPLE options\n"
+                
+            elif question['type'] == 'rating':
+                prompt += f"Rating range: {question.get('min_rating', 1)} - {question.get('max_rating', 5)}\n"
+                prompt += "Please provide a rating within the range\n"
+                
+            elif question['type'] == 'matrix':
+                prompt += "Rows: " + ", ".join(question['rows']) + "\n"
+                prompt += "Columns: " + ", ".join(question['columns']) + "\n"
+                prompt += "Please select ONE column option for EACH row\n"
+                
+            elif question['type'] == 'text':
+                prompt += "Please provide a text response\n"
+                
+            elif question['type'] == 'boolean':
+                prompt += "Options: Yes, No\n"
+                prompt += "Please select either Yes or No\n"
+                
+            prompt += "\nAnswer: [Your response here]\n\n---\n\n"
+            question_count += 1
+            
+    # 添加总结提示
+    prompt += """Please ensure:
+1. All required questions are answered
+2. Responses match the question type requirements
+3. Answers are clear and specific
+
+Format your responses exactly as requested above."""
+    
+    return prompt

pycityagent/workflow/tool.py

@@ -1,6 +1,3 @@
-import asyncio
-import logging
-from copy import deepcopy
 from typing import Any, Callable, Dict, List, Optional, Union
 
 from ..agent import Agent

{pycityagent-2.0.0a13.dist-info → pycityagent-2.0.0a15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycityagent
-Version: 2.0.0a13
+Version: 2.0.0a15
 Summary: LLM-based城市环境agent构建库
 License: MIT
 Author: Yuwei Yan
@@ -20,6 +20,7 @@ Requires-Dist: aiohttp (==3.10.10)
 Requires-Dist: aiomqtt (>=2.3.0,<3.0.0)
 Requires-Dist: citystreetview (==1.2.4)
 Requires-Dist: dashscope (==1.14.1)
+Requires-Dist: fastavro (>=1.10.0,<2.0.0)
 Requires-Dist: geojson (==3.1.0)
 Requires-Dist: gradio (>=5.7.1,<6.0.0)
 Requires-Dist: grpcio (==1.67.1)
@@ -28,6 +29,7 @@ Requires-Dist: mosstool (==1.0.24)
 Requires-Dist: networkx (==3.2.1)
 Requires-Dist: numpy (>=1.20.0,<2.0.0)
 Requires-Dist: openai (>=1.58.1,<2.0.0)
+Requires-Dist: pandavro (>=1.8.0,<2.0.0)
 Requires-Dist: poetry (>=1.2.2)
 Requires-Dist: protobuf (<=4.24.0)
 Requires-Dist: pycitydata (==1.0.0)

{pycityagent-2.0.0a13.dist-info → pycityagent-2.0.0a15.dist-info}/RECORD

@@ -1,7 +1,7 @@
-pycityagent/__init__.py,sha256=n56bWkAUEcvjDsb7LcJpaGjlrriSKPnR0yBhwRfEYBA,212
-pycityagent/agent.py,sha256=7BCHK5OH1TvW16JLiYR0iWAiWGBuP4eJsiGU1y7exbY,18417
+pycityagent/__init__.py,sha256=EDxt3Su3lH1IMh9suNw7GeGL7UrXeWiZTw5KWNznDzc,637
+pycityagent/agent.py,sha256=fcuKX6FtMzjNP8lVep9pG-9KHzHQwJ8IymJbmLKMfu0,23109
 pycityagent/economy/__init__.py,sha256=aonY4WHnx-6EGJ4WKrx4S-2jAkYNLtqUA04jp6q8B7w,75
-pycityagent/economy/econ_client.py,sha256=qQb_kZneEXGBRaS_y5Jdoi95I8GyjKEsDSC4s6V6R7w,10829
+pycityagent/economy/econ_client.py,sha256=wcuNtcpkSijJwNkt2mXw3SshYy4SBy6qbvJ0VQ7Aovo,10854
 pycityagent/environment/__init__.py,sha256=awHxlOud-btWbk0FCS4RmGJ13W84oVCkbGfcrhKqihA,240
 pycityagent/environment/interact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycityagent/environment/interact/interact.py,sha256=ifxPPzuHeqLHIZ_6zvfXMoBOnBsXNIP4bYp7OJ7pnEQ,6588
@@ -21,7 +21,7 @@ pycityagent/environment/sim/person_service.py,sha256=nIvOsoBoqOTDYtsiThg07-4ZBgk
 pycityagent/environment/sim/road_service.py,sha256=phKTwTyhc_6Ht2mddEXpdENfl-lRXIVY0CHAlw1yHjI,1264
 pycityagent/environment/sim/sim_env.py,sha256=HI1LcS_FotDKQ6vBnx0e49prXSABOfA20aU9KM-ZkCY,4625
 pycityagent/environment/sim/social_service.py,sha256=6Iqvq6dz8H2jhLLdtaITc6Js9QnQw-Ylsd5AZgUj3-E,1993
-pycityagent/environment/simulator.py,sha256=fQv6D_1OnhUrKTsnnah3wnm9ec8LE1phxRhK1K93Zyg,12466
+pycityagent/environment/simulator.py,sha256=K7IyhiGC9BxanW28bpML4M0YREdMp1h7yMoWBlbf3RY,12504
 pycityagent/environment/utils/__init__.py,sha256=1m4Q1EfGvNpUsa1bgQzzCyWhfkpElnskNImjjFD3Znc,237
 pycityagent/environment/utils/base64.py,sha256=hoREzQo3FXMN79pqQLO2jgsDEvudciomyKii7MWljAM,374
 pycityagent/environment/utils/const.py,sha256=3RMNy7_bE7-23K90j9DFW_tWEzu8s7hSTgKbV-3BFl4,5327
@@ -36,35 +36,34 @@ pycityagent/llm/llm.py,sha256=BtxBvPK4tb8QlZIfxO5XJ73lKXwF8L31LqVbejWB8eo,15121
 pycityagent/llm/llmconfig.py,sha256=4Ylf4OFSBEFy8jrOneeX0HvPhWEaF5jGvy1HkXK08Ro,436
 pycityagent/llm/utils.py,sha256=hoNPhvomb1u6lhFX0GctFipw74hVKb7bvUBDqwBzBYw,160
 pycityagent/memory/__init__.py,sha256=Hs2NhYpIG-lvpwPWwj4DydB1sxtjz7cuA4iDAzCXnjI,243
-pycityagent/memory/const.py,sha256=m9AidLs7Zu28StpvYetclqx-1qQcy3bYvwagcXB3a04,913
-pycityagent/memory/memory.py,sha256=sDbaqr1Koqf_9joMtG9PmmVxJZ6Rq7nAZO6EO0OdVgo,18148
-pycityagent/memory/memory_base.py,sha256=bd2q0qNu5hCRd2u4cPxE3bBA2OaoAD1oR4-vbRdbd_s,5594
+pycityagent/memory/const.py,sha256=6zpJPJXWoH9-yf4RARYYff586agCoud9BRn7sPERB1g,932
+pycityagent/memory/memory.py,sha256=FjKVL_MgNBnSc0sox2tuxLqXg9_MQQr9vYdRDHMdDL4,18183
+pycityagent/memory/memory_base.py,sha256=euKZRCs4dbcKxjlZzpLCTnH066DAtRjj5g1JFKD40qQ,5633
 pycityagent/memory/profile.py,sha256=s4LnxSPGSjIGZXHXkkd8mMa6uYYZrytgyQdWjcaqGf4,5182
 pycityagent/memory/self_define.py,sha256=poPiexNhOLq_iTgK8s4mK_xoL_DAAcB8kMvInj7iE5E,5179
 pycityagent/memory/state.py,sha256=5W0c1yJ-aaPpE74B2LEcw3Ygpm77tyooHv8NylyrozE,5113
 pycityagent/memory/utils.py,sha256=wLNlNlZ-AY9VB8kbUIy0UQSYh26FOQABbhmKQkit5o8,850
 pycityagent/message/__init__.py,sha256=TCjazxqb5DVwbTu1fF0sNvaH_EPXVuj2XQ0p6W-QCLU,55
-pycityagent/message/messager.py,sha256=Iv4pK83JvHAQSZyGNACryPBey2wRoiok3Hb1eIwHbww,2506
+pycityagent/message/messager.py,sha256=W_OVlNGcreHSBf6v-DrEnfNCXExB78ySr0w26MSncfU,2541
 pycityagent/simulation/__init__.py,sha256=jYaqaNpzM5M_e_ykISS_M-mIyYdzJXJWhgpfBpA6l5k,111
-pycityagent/simulation/agentgroup.py,sha256=fM93ydsuZ_w8YVnHnX1tWskWVYmt6sztKjzgDXcg1Xw,6106
-pycityagent/simulation/interview.py,sha256=S2uv8MFCB4-u_4Q202VFoPJOIleqpKK9Piju0BDSb_0,1158
-pycityagent/simulation/simulation.py,sha256=4BHavfSHPM81Y5Su-MzzBYM0sR9cHRk4YOv-VkRkzrA,11595
-pycityagent/simulation/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
-pycityagent/simulation/survey/manager.py,sha256=Tqini-4-uBZDsChVVL4ezlgXatnrxAfvceAZZRP8E48,2062
-pycityagent/simulation/survey/models.py,sha256=sY4OrrG1h9iBnjBsyDage4T3mUFPBHHZQe-ORtwSjKc,1305
-pycityagent/simulation/ui/__init__.py,sha256=4aevYl09KSku2NEpFZhyFNxI1qlswbrAFwhF_YhxFTo,62
-pycityagent/simulation/ui/interface.py,sha256=cSQRQhuA5Gj-X7anuN9tmRpG6-I8QYOuswjIWUeSvio,21636
-pycityagent/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycityagent/simulation/agentgroup.py,sha256=JwfssUtVrOgSnJCan4jcIcSHLjWBCwYxqOPT-AXA2sE,12514
+pycityagent/simulation/simulation.py,sha256=G68P1EJ3JceA3zID2O6AGd_KdhhYy5XVZVUgkfJHypc,18897
+pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
+pycityagent/survey/manager.py,sha256=N4-Q8vve4L-PLaFikAgVB4Z8BNyFDEd6WjEk3AyMTgs,1764
+pycityagent/survey/models.py,sha256=Z2gHEazQRj0TkTz5qbh4Uy_JrU_FZGWpOLwjN0RoUrY,3547
+pycityagent/utils/__init__.py,sha256=xXEMhVfFeOJUXjczaHv9DJqYNp57rc6FibtS7CfrVbA,305
+pycityagent/utils/avro_schema.py,sha256=DHM3bOo8m0dJf8oSwyOWzVeXrH6OERmzA_a5vS4So4M,4255
 pycityagent/utils/decorators.py,sha256=Gk3r41hfk6awui40tbwpq3C7wC7jHaRmLRlcJFlLQCE,3160
 pycityagent/utils/parsers/__init__.py,sha256=AN2xgiPxszWK4rpX7zrqRsqNwfGF3WnCA5-PFTvbaKk,281
 pycityagent/utils/parsers/code_block_parser.py,sha256=Cs2Z_hm9VfNCpPPll1TwteaJF-HAQPs-3RApsOekFm4,1173
 pycityagent/utils/parsers/json_parser.py,sha256=FZ3XN1g8z4Dr2TFraUOoah1oQcze4fPd2m01hHoX0Mo,2917
 pycityagent/utils/parsers/parser_base.py,sha256=k6DVqwAMK3jJdOP4IeLE-aFPm3V2F-St5qRBuRdx4aU,1742
+pycityagent/utils/survey_util.py,sha256=Be9nptmu2JtesFNemPgORh_2GsN7rcDYGQS9Zfvc5OI,2169
 pycityagent/workflow/__init__.py,sha256=EyCcjB6LyBim-5iAOPe4m2qfvghEPqu1ZdGfy4KPeZ8,551
 pycityagent/workflow/block.py,sha256=6EmiRMLdOZC1wMlmLMIjfrp9TuiI7Gw4s3nnXVMbrnw,6031
 pycityagent/workflow/prompt.py,sha256=tY69nDO8fgYfF_dOA-iceR8pAhkYmCqoox8uRPqEuGY,2956
-pycityagent/workflow/tool.py,sha256=wD9WZ5rma6HCKugtHTwbShNE0f-Rjlwvn_1be3fCAsk,6682
+pycityagent/workflow/tool.py,sha256=zMvz3BV4QBs5TqyQ3ziJxj4pCfL2uqUI3A1FbT1gd3Q,6626
 pycityagent/workflow/trigger.py,sha256=t5X_i0WtL32bipZSsq_E3UUyYYudYLxQUpvxbgClp2s,5683
-pycityagent-2.0.0a13.dist-info/METADATA,sha256=MdIVm_sOeGM99wtIKpcDEt6zt3Gp-loHI5MxGTz_6og,7622
-pycityagent-2.0.0a13.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pycityagent-2.0.0a13.dist-info/RECORD,,
+pycityagent-2.0.0a15.dist-info/METADATA,sha256=8ONKHaTIPOGja6FKJDE9tZ-pLZL2aeXQufB-LWkElR8,7705
+pycityagent-2.0.0a15.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pycityagent-2.0.0a15.dist-info/RECORD,,

pycityagent/simulation/interview.py

@@ -1,40 +0,0 @@
-from dataclasses import dataclass
-from datetime import datetime
-from typing import List, Optional
-
-
-@dataclass
-class InterviewRecord:
-    """采访记录"""
-
-    timestamp: datetime
-    agent_name: str
-    question: str
-    response: str
-    blocking: bool
-
-
-class InterviewManager:
-    """采访管理器"""
-
-    def __init__(self):
-        self._history: List[InterviewRecord] = []
-
-    def add_record(self, agent_name: str, question: str, response: str, blocking: bool):
-        """添加采访记录"""
-        record = InterviewRecord(
-            timestamp=datetime.now(),
-            agent_name=agent_name,
-            question=question,
-            response=response,
-            blocking=blocking,
-        )
-        self._history.append(record)
-
-    def get_agent_history(self, agent_name: str) -> List[InterviewRecord]:
-        """获取指定智能体的采访历史"""
-        return [r for r in self._history if r.agent_name == agent_name]
-
-    def get_recent_history(self, limit: int = 10) -> List[InterviewRecord]:
-        """获取最近的采访记录"""
-        return sorted(self._history, key=lambda x: x.timestamp, reverse=True)[:limit]

pycityagent/simulation/survey/manager.py

@@ -1,68 +0,0 @@
-from typing import List, Dict, Optional
-from datetime import datetime
-import uuid
-import json
-from .models import Survey, Question, QuestionType
-
-
-class SurveyManager:
-    def __init__(self):
-        self._surveys: Dict[str, Survey] = {}
-
-    def create_survey(
-        self, title: str, description: str, questions: List[dict]
-    ) -> Survey:
-        """创建新问卷"""
-        survey_id = str(uuid.uuid4())
-
-        # 转换问题数据
-        survey_questions = []
-        for q in questions:
-            question = Question(
-                content=q["content"],
-                type=QuestionType(q["type"]),
-                required=q.get("required", True),
-                options=q.get("options", []),
-                min_rating=q.get("min_rating", 1),
-                max_rating=q.get("max_rating", 5),
-            )
-            survey_questions.append(question)
-
-        survey = Survey(
-            id=survey_id,
-            title=title,
-            description=description,
-            questions=survey_questions,
-        )
-
-        self._surveys[survey_id] = survey
-        return survey
-
-    def get_survey(self, survey_id: str) -> Optional[Survey]:
-        """获取指定问卷"""
-        return self._surveys.get(survey_id)
-
-    def get_all_surveys(self) -> List[Survey]:
-        """获取所有问卷"""
-        return list(self._surveys.values())
-
-    def add_response(self, survey_id: str, agent_name: str, response: dict) -> bool:
-        """添加问卷回答"""
-        survey = self.get_survey(survey_id)
-        if not survey:
-            return False
-
-        survey.responses[agent_name] = {"timestamp": datetime.now(), **response}
-        return True
-
-    def export_results(self, survey_id: str) -> str:
-        """导出问卷结果"""
-        survey = self.get_survey(survey_id)
-        if not survey:
-            return json.dumps({"error": "问卷不存在"})
-
-        return json.dumps(
-            {"survey": survey.to_dict(), "responses": survey.responses},
-            ensure_ascii=False,
-            indent=2,
-        )

pycityagent/simulation/survey/models.py

@@ -1,52 +0,0 @@
-from dataclasses import dataclass, field
-from typing import List, Dict, Optional
-from datetime import datetime
-from enum import Enum
-import uuid
-
-
-class QuestionType(Enum):
-    TEXT = "文本"
-    SINGLE_CHOICE = "单选"
-    MULTIPLE_CHOICE = "多选"
-    RATING = "评分"
-    LIKERT = "李克特量表"
-
-
-@dataclass
-class Question:
-    content: str
-    type: QuestionType
-    required: bool = True
-    options: List[str] = field(default_factory=list)
-    min_rating: int = 1
-    max_rating: int = 5
-
-    def to_dict(self) -> dict:
-        return {
-            "content": self.content,
-            "type": self.type.value,
-            "required": self.required,
-            "options": self.options,
-            "min_rating": self.min_rating,
-            "max_rating": self.max_rating,
-        }
-
-
-@dataclass
-class Survey:
-    id: str
-    title: str
-    description: str
-    questions: List[Question]
-    responses: Dict[str, dict] = field(default_factory=dict)
-    created_at: datetime = field(default_factory=datetime.now)
-
-    def to_dict(self) -> dict:
-        return {
-            "id": self.id,
-            "title": self.title,
-            "description": self.description,
-            "questions": [q.to_dict() for q in self.questions],
-            "response_count": len(self.responses),
-        }

pycityagent/simulation/ui/__init__.py

@@ -1,3 +0,0 @@
-from .interface import InterviewUI
-
-__all__ = ["InterviewUI"]