vip11017 commited on
Commit
41765be
·
1 Parent(s): fa5a94c

added prcing_plan inside dict

Browse files
app/ingestion/models.py CHANGED
@@ -6,6 +6,7 @@ class ChatbotIngest(BaseModel):
6
  submission_id: str
7
  company_name: str
8
  chatbot_name: Optional[str] = None
 
9
  website_url: HttpUrl
10
  chatbot_purpose: List[str] = []
11
  additional_content: Optional[List[str]] = []
 
6
  submission_id: str
7
  company_name: str
8
  chatbot_name: Optional[str] = None
9
+ pricing_plan: Optional[str]=None
10
  website_url: HttpUrl
11
  chatbot_purpose: List[str] = []
12
  additional_content: Optional[List[str]] = []
app/ingestion/utils.py CHANGED
@@ -32,6 +32,8 @@ def extract_tally_payload(payload: dict):
32
  # Submission ID
33
  submission_id = payload.get("data", {}).get("submissionId")
34
 
 
 
35
  # Company info
36
  company_name = key_map.get("question_kyp5Dd", {}).get("value")
37
  chatbot_name = key_map.get("question_vAp47X", {}).get("value") or f"{company_name} Assistant"
@@ -71,6 +73,7 @@ def extract_tally_payload(payload: dict):
71
  "submission_id": submission_id,
72
  "company_name": company_name,
73
  "chatbot_name": chatbot_name,
 
74
  "website_url": website_url,
75
  "chatbot_purpose": chatbot_purpose,
76
  "additional_content": additional_content,
 
32
  # Submission ID
33
  submission_id = payload.get("data", {}).get("submissionId")
34
 
35
+ pricing_plan = key_map.get("question_K6rZP8_15f06788-71a0-4a4a-903e-68944b6cb967", {}).get("value") or None
36
+
37
  # Company info
38
  company_name = key_map.get("question_kyp5Dd", {}).get("value")
39
  chatbot_name = key_map.get("question_vAp47X", {}).get("value") or f"{company_name} Assistant"
 
73
  "submission_id": submission_id,
74
  "company_name": company_name,
75
  "chatbot_name": chatbot_name,
76
+ "pricing_plan": pricing_plan,
77
  "website_url": website_url,
78
  "chatbot_purpose": chatbot_purpose,
79
  "additional_content": additional_content,