41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
|
import requests
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
# API 地址
|
|||
|
|
API_URL = "http://0.0.0.0:5163/embedding/"
|
|||
|
|
|
|||
|
|
# 要发送的数据（JSON 格式）
|
|||
|
|
data = {
|
|||
|
|
"sentences": "土豆" # 单句示例，可改为 ["土豆", "西红柿"] 处理多句
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 发送 POST 请求
|
|||
|
|
try:
|
|||
|
|
# 设置请求头，明确指定内容类型为 JSON
|
|||
|
|
headers = {"Content-Type": "application/json"}
|
|||
|
|
|
|||
|
|
# 使用 POST 方法发送请求，data 需要序列化为 JSON 字符串
|
|||
|
|
response = requests.post(
|
|||
|
|
url=API_URL,
|
|||
|
|
data=json.dumps(data), # 将字典转换为 JSON 字符串
|
|||
|
|
headers=headers
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 检查响应状态码
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
# 解析返回的 JSON 数据
|
|||
|
|
result = response.json()
|
|||
|
|
print("API Response:", result)
|
|||
|
|
|
|||
|
|
# 根据返回的 code 判断结果
|
|||
|
|
if result.get("code") == 200:
|
|||
|
|
embeddings = result.get("data")
|
|||
|
|
print("Embeddings:", embeddings)
|
|||
|
|
else:
|
|||
|
|
print("Error from API:", result.get("msg"))
|
|||
|
|
else:
|
|||
|
|
print(f"Request failed with status code: {response.status_code}")
|
|||
|
|
print("Response text:", response.text)
|
|||
|
|
|
|||
|
|
except requests.exceptions.RequestException as e:
|
|||
|
|
print(f"Error connecting to API: {e}")
|