embedchain
146 строк · 3.3 Кб
1{
2"cells": [
3{
4"cell_type": "markdown",
5"metadata": {
6"id": "b02n_zJ_hl3d"
7},
8"source": [
9"## Cookbook for using PineconeDB with Embedchain"
10]
11},
12{
13"cell_type": "markdown",
14"metadata": {
15"id": "gyJ6ui2vhtMY"
16},
17"source": [
18"### Step-1: Install embedchain package"
19]
20},
21{
22"cell_type": "code",
23"execution_count": null,
24"metadata": {
25"id": "-NbXjAdlh0vJ"
26},
27"outputs": [],
28"source": [
29"!pip install embedchain pinecone-client pinecone-text"
30]
31},
32{
33"cell_type": "markdown",
34"metadata": {
35"id": "nGnpSYAAh2bQ"
36},
37"source": [
38"### Step-2: Set environment variables needed for Pinecone\n",
39"\n",
40"You can find this env variable on your [OpenAI dashboard](https://platform.openai.com/account/api-keys) and [Pinecone dashboard](https://app.pinecone.io/)."
41]
42},
43{
44"cell_type": "code",
45"execution_count": null,
46"metadata": {
47"id": "0fBdQ9GAiRvK"
48},
49"outputs": [],
50"source": [
51"import os\n",
52"from embedchain import App\n",
53"\n",
54"os.environ[\"OPENAI_API_KEY\"] = \"sk-xxx\"\n",
55"os.environ[\"PINECONE_API_KEY\"] = \"xxx\"\n",
56"os.environ[\"PINECONE_ENV\"] = \"xxx\""
57]
58},
59{
60"cell_type": "markdown",
61"metadata": {
62"id": "PGt6uPLIi1CS"
63},
64"source": [
65"### Step-3 Create embedchain app and define your config"
66]
67},
68{
69"cell_type": "code",
70"execution_count": null,
71"metadata": {
72"id": "Amzxk3m-i3tD"
73},
74"outputs": [],
75"source": [
76"app = App.from_config(config={\n",
77" \"provider\": \"pinecone\",\n",
78" \"config\": {\n",
79" \"metric\": \"cosine\",\n",
80" \"vector_dimension\": 768,\n",
81" \"collection_name\": \"pc-index\"\n",
82" }\n",
83"})"
84]
85},
86{
87"cell_type": "markdown",
88"metadata": {
89"id": "XNXv4yZwi7ef"
90},
91"source": [
92"### Step-4: Add data sources to your app"
93]
94},
95{
96"cell_type": "code",
97"execution_count": null,
98"metadata": {
99"id": "Sn_0rx9QjIY9"
100},
101"outputs": [],
102"source": [
103"app.add(\"https://www.forbes.com/profile/elon-musk\")"
104]
105},
106{
107"cell_type": "markdown",
108"metadata": {
109"id": "_7W6fDeAjMAP"
110},
111"source": [
112"### Step-5: All set. Now start asking questions related to your data"
113]
114},
115{
116"cell_type": "code",
117"execution_count": null,
118"metadata": {
119"id": "cvIK7dWRjN_f"
120},
121"outputs": [],
122"source": [
123"while(True):\n",
124" question = input(\"Enter question: \")\n",
125" if question in ['q', 'exit', 'quit']:\n",
126" break\n",
127" answer = app.query(question)\n",
128" print(answer)"
129]
130}
131],
132"metadata": {
133"colab": {
134"provenance": []
135},
136"kernelspec": {
137"display_name": "Python 3",
138"name": "python3"
139},
140"language_info": {
141"name": "python"
142}
143},
144"nbformat": 4,
145"nbformat_minor": 0
146}
147