Spaces:
Sleeping
Sleeping
added prcing_plan inside dict
Browse files- app/ingestion/models.py +1 -0
- app/ingestion/utils.py +3 -0
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,
|