李顺东
b66bac7ca8
- Add wxauto package with WeChat UI automation and message handling capabilities - Implement job_extractor.py for automated job posting extraction from WeChat groups - Add job_extractor_gui.py providing graphical interface for job extraction tool - Create comprehensive documentation in Chinese covering GUI usage, multi-group support, and quick start guides - Add build configuration files (build_exe.py, build_exe.spec) for packaging as standalone executable - Include utility scripts for WeChat interaction (auto_send_msg.py, get_history.py, receive_file_transfer.py) - Add project configuration files (pyproject.toml, setup.cfg, requirements.txt) - Include test files (test_api.py, test_com_fix.py) for API and compatibility validation - Add Apache 2.0 LICENSE and comprehensive README documentation - Configure .gitignore to exclude build artifacts, logs, and temporary files
142 lines
3.9 KiB
Python
142 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
测试百炼API连接和岗位信息提取功能
|
||
"""
|
||
import json
|
||
import requests
|
||
|
||
# API配置
|
||
BAILIAN_API_URL = "https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation"
|
||
API_KEY = "sk-46cb053d75eb4ad88713917ba0f1c81a"
|
||
|
||
# 测试消息样例
|
||
TEST_MESSAGES = [
|
||
"""
|
||
【招聘】Python后端开发工程师
|
||
公司：北京某互联网科技有限公司
|
||
地点：北京市海淀区中关村
|
||
薪资：20K-35K
|
||
岗位职责：
|
||
1. 负责公司核心业务系统的后端开发
|
||
2. 参与系统架构设计和优化
|
||
要求：3年以上Python开发经验，熟悉Django/Flask
|
||
联系人：李经理
|
||
微信：tech_hr_2024
|
||
""",
|
||
|
||
"""
|
||
急招前端开发！！！
|
||
坐标：上海浦东
|
||
月薪：15-25k
|
||
我们公司是做电商的，需要熟悉Vue3和React的前端
|
||
有意者联系张总 18612345678
|
||
""",
|
||
|
||
"""
|
||
今天天气真好，大家周末愉快！
|
||
"""
|
||
]
|
||
|
||
|
||
def test_extract(message_content):
|
||
"""测试提取功能"""
|
||
print("=" * 80)
|
||
print("测试消息:")
|
||
print(message_content.strip())
|
||
print("-" * 80)
|
||
|
||
prompt = f"""请从以下消息中提取招聘岗位信息，并以JSON格式返回。如果消息不包含招聘信息，返回空对象。
|
||
|
||
要提取的字段：
|
||
- job_name: 工作名称
|
||
- job_description: 工作描述
|
||
- job_location: 工作地点
|
||
- salary_min: 月薪最低（数字，单位：元）
|
||
- salary_max: 月薪最高（数字，单位：元）
|
||
- company_name: 公司名称
|
||
- contact_person: 联系人
|
||
- contact_info: 联系方式（电话/微信等）
|
||
|
||
消息内容：
|
||
{message_content}
|
||
|
||
请直接返回JSON格式，不要包含其他说明文字。"""
|
||
|
||
headers = {
|
||
"Content-Type": "application/json",
|
||
"Authorization": f"Bearer {API_KEY}"
|
||
}
|
||
|
||
payload = {
|
||
"model": "qwen-plus",
|
||
"input": {
|
||
"messages": [
|
||
{
|
||
"role": "system",
|
||
"content": "你是一个专业的招聘信息提取助手，擅长从文本中提取结构化的岗位信息。"
|
||
},
|
||
{
|
||
"role": "user",
|
||
"content": prompt
|
||
}
|
||
]
|
||
},
|
||
"parameters": {
|
||
"result_format": "message"
|
||
}
|
||
}
|
||
|
||
try:
|
||
print("正在调用API...")
|
||
response = requests.post(BAILIAN_API_URL, headers=headers, json=payload, timeout=30)
|
||
response.raise_for_status()
|
||
result = response.json()
|
||
|
||
if "output" in result and "choices" in result["output"]:
|
||
content = result["output"]["choices"][0]["message"]["content"]
|
||
print("API返回:")
|
||
print(content)
|
||
print("-" * 80)
|
||
|
||
# 尝试解析JSON
|
||
try:
|
||
job_data = json.loads(content)
|
||
if job_data and any(job_data.values()):
|
||
print("✓ 提取成功:")
|
||
print(json.dumps(job_data, ensure_ascii=False, indent=2))
|
||
else:
|
||
print("× 未提取到有效岗位信息")
|
||
except json.JSONDecodeError:
|
||
print(f"× JSON解析失败")
|
||
else:
|
||
print(f"× API返回格式异常: {result}")
|
||
|
||
except requests.exceptions.RequestException as e:
|
||
print(f"× API请求失败: {e}")
|
||
except Exception as e:
|
||
print(f"× 发生错误: {e}")
|
||
|
||
print()
|
||
|
||
|
||
def main():
|
||
print("百炼API岗位信息提取测试")
|
||
print("=" * 80)
|
||
print(f"API地址: {BAILIAN_API_URL}")
|
||
print(f"API密钥: {API_KEY[:20]}...")
|
||
print()
|
||
|
||
for i, msg in enumerate(TEST_MESSAGES, 1):
|
||
print(f"\n测试用例 {i}/{len(TEST_MESSAGES)}")
|
||
test_extract(msg)
|
||
|
||
if i < len(TEST_MESSAGES):
|
||
input("按回车继续下一个测试...")
|
||
|
||
print("=" * 80)
|
||
print("测试完成")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|