From 40a39a0f1a4203e853a89454109e95768b0bb6c6 Mon Sep 17 00:00:00 2001
From: mamamiyear <mamamiyear@163.com>
Date: Mon, 27 Oct 2025 11:44:01 +0800
Subject: [PATCH] feat: people add new filed 'contact'

---
 src/ai/agent.py             |  1 -
 src/models/people.py        | 10 ++++------
 src/storage/people_store.py |  3 +++
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/ai/agent.py b/src/ai/agent.py
index f10d61f..e508a4c 100644
--- a/src/ai/agent.py
+++ b/src/ai/agent.py
@@ -25,7 +25,6 @@ class ExtractPeopleAgent(BaseAgent):
                 "性别 gender\n"
                 "年龄 age\n"
                 "身高(cm) height\n"
-                # "体重(kg) weight\n"
                 "婚姻状况 marital_status\n"
                 "择偶要求 match_requirement\n"
                 "以上信息需要严格按照 JSON 格式输出 字段名与条目中英文保持一致。\n"
diff --git a/src/models/people.py b/src/models/people.py
index f30eb48..5b96a9c 100644
--- a/src/models/people.py
+++ b/src/models/people.py
@@ -10,6 +10,8 @@ class People:
     id: str
     # 姓名
     name: str
+    # 联系人
+    contact: str
     # 性别
     gender: str
     # 年龄
@@ -38,10 +40,10 @@ class People:
         # 初始化所有属性，从kwargs中获取值，如果不存在则设置默认值
         self.id = kwargs.get('id', '') if kwargs.get('id', '') is not None else ''
         self.name = kwargs.get('name', '') if kwargs.get('name', '') is not None else ''
+        self.contact = kwargs.get('contact', '') if kwargs.get('contact', '') is not None else ''
         self.gender = kwargs.get('gender', '') if kwargs.get('gender', '') is not None else ''
         self.age = kwargs.get('age', 0) if kwargs.get('age', 0) is not None else 0
         self.height = kwargs.get('height', 0) if kwargs.get('height', 0) is not None else 0
-        # self.weight = kwargs.get('weight', 0) if kwargs.get('weight', 0) is not None else 0
         self.marital_status = kwargs.get('marital_status', '') if kwargs.get('marital_status', '') is not None else ''
         self.match_requirement = kwargs.get('match_requirement', '') if kwargs.get('match_requirement', '') is not None else ''
         self.introduction = kwargs.get('introduction', {}) if kwargs.get('introduction', {}) is not None else {}
@@ -68,10 +70,10 @@ class People:
         return {
             'id': self.id,
             'name': self.name,
+            'contact': self.contact,
             'gender': self.gender,
             'age': self.age,
             'height': self.height,
-            # 'weight': self.weight,
             'marital_status': self.marital_status,
             'match_requirement': self.match_requirement,
             'introduction': self.introduction,
@@ -86,9 +88,7 @@ class People:
             'gender': self.gender,
             'age': self.age,
             'height': self.height,
-            # 'weight': self.weight,
             'marital_status': self.marital_status,
-            # 'match_requirement': self.match_requirement,
         }
         logging.info(f"people meta: {meta}")
         return meta
@@ -102,8 +102,6 @@ class People:
             doc.append(f"年龄: {self.age}")
         if self.height:
             doc.append(f"身高: {self.height}cm")
-        # if self.weight:
-            # doc.append(f"体重: {self.weight}kg")
         if self.marital_status:
             doc.append(f"婚姻状况: {self.marital_status}")
         if self.match_requirement:
diff --git a/src/storage/people_store.py b/src/storage/people_store.py
index 67768ea..2086e32 100644
--- a/src/storage/people_store.py
+++ b/src/storage/people_store.py
@@ -19,6 +19,7 @@ class PeopleORM(Base):
     __tablename__ = 'peoples'
     id = Column(String(36), primary_key=True)
     name = Column(String(255), index=True)
+    contact = Column(String(255), index=True)
     gender = Column(String(10))
     age = Column(Integer)
     height = Column(Integer)
@@ -33,6 +34,7 @@ class PeopleORM(Base):
         import json
         self.id = people.id
         self.name = people.name
+        self.contact = people.contact
         self.gender = people.gender
         self.age = people.age
         self.height = people.height
@@ -47,6 +49,7 @@ class PeopleORM(Base):
         people = People()
         people.id = self.id
         people.name = self.name
+        people.contact = self.contact
         people.gender = self.gender
         people.age = self.age
         people.height = self.height