feat: rag

15vector.py

@@ -0,0 +1,19 @@
+# 使用豆包来向量化文本
+
+import os
+from volcenginesdkarkruntime import Ark
+
+# 初始化客户端
+client = Ark(
+    # 从环境变量中读取您的方舟API Key
+    api_key=os.environ.get("ARK_API_KEY", "79b39c58-56db-4d8a-a8f8-84b95fca08db"),
+    base_url="https://ark.cn-beijing.volces.com/api/v3",
+)
+response = client.embeddings.create(
+    model="doubao-embedding-text-240715",
+    input="Function Calling 是一种将大模型与外部工具和 API 相连的关键功能",
+    encoding_format="float",
+)
+# 打印结果
+print(f"向量维度: {len(response.data[0].embedding)}")
+print(f"前10维向量: {response.data[0].embedding[:10]}")