pycityagent 2.0.0a12__py3-none-any.whl → 2.0.0a14__py3-none-any.whl

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"]