embedchain

1
{
2
  "cells": [
3
    {
4
      "cell_type": "markdown",
5
      "metadata": {
6
        "id": "b02n_zJ_hl3d"
7
      },
8
      "source": [
9
        "## Cookbook for using VertexAI 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
        "colab": {
26
          "base_uri": "https://localhost:8080/"
27
        },
28
        "id": "-NbXjAdlh0vJ",
29
        "outputId": "eb9be5b6-dc81-43d2-d515-df8f0116be11"
30
      },
31
      "outputs": [],
32
      "source": [
33
        "!pip install embedchain[vertexai]"
34
      ]
35
    },
36
    {
37
      "cell_type": "markdown",
38
      "metadata": {
39
        "id": "nGnpSYAAh2bQ"
40
      },
41
      "source": [
42
        "### Step-2: Set VertexAI related environment variables\n",
43
        "\n",
44
        "You can find `OPENAI_API_KEY` on your [OpenAI dashboard](https://platform.openai.com/account/api-keys)."
45
      ]
46
    },
47
    {
48
      "cell_type": "code",
49
      "execution_count": null,
50
      "metadata": {
51
        "id": "0fBdQ9GAiRvK"
52
      },
53
      "outputs": [],
54
      "source": [
55
        "import os\n",
56
        "from embedchain import App\n",
57
        "\n",
58
        "os.environ[\"OPENAI_API_KEY\"] = \"sk-xxx\""
59
      ]
60
    },
61
    {
62
      "cell_type": "markdown",
63
      "metadata": {
64
        "id": "PGt6uPLIi1CS"
65
      },
66
      "source": [
67
        "### Step-3 Create embedchain app and define your config"
68
      ]
69
    },
70
    {
71
      "cell_type": "code",
72
      "execution_count": null,
73
      "metadata": {
74
        "colab": {
75
          "base_uri": "https://localhost:8080/",
76
          "height": 582
77
        },
78
        "id": "Amzxk3m-i3tD",
79
        "outputId": "5084b6ea-ec20-4281-9f36-e21e93c17475"
80
      },
81
      "outputs": [],
82
      "source": [
83
        "app = App.from_config(config={\n",
84
        "    \"llm\": {\n",
85
        "        \"provider\": \"vertexai\",\n",
86
        "        \"config\": {\n",
87
        "            \"model\": \"chat-bison\",\n",
88
        "            \"temperature\": 0.5,\n",
89
        "            \"max_tokens\": 1000,\n",
90
        "            \"stream\": False\n",
91
        "        }\n",
92
        "    },\n",
93
        "    \"embedder\": {\n",
94
        "        \"provider\": \"vertexai\",\n",
95
        "        \"config\": {\n",
96
        "            \"model\": \"textembedding-gecko\"\n",
97
        "        }\n",
98
        "    }\n",
99
        "})"
100
      ]
101
    },
102
    {
103
      "cell_type": "markdown",
104
      "metadata": {
105
        "id": "XNXv4yZwi7ef"
106
      },
107
      "source": [
108
        "### Step-4: Add data sources to your app"
109
      ]
110
    },
111
    {
112
      "cell_type": "code",
113
      "execution_count": null,
114
      "metadata": {
115
        "id": "Sn_0rx9QjIY9"
116
      },
117
      "outputs": [],
118
      "source": [
119
        "app.add(\"https://www.forbes.com/profile/elon-musk\")"
120
      ]
121
    },
122
    {
123
      "cell_type": "markdown",
124
      "metadata": {
125
        "id": "_7W6fDeAjMAP"
126
      },
127
      "source": [
128
        "### Step-5: All set. Now start asking questions related to your data"
129
      ]
130
    },
131
    {
132
      "cell_type": "code",
133
      "execution_count": null,
134
      "metadata": {
135
        "id": "cvIK7dWRjN_f"
136
      },
137
      "outputs": [],
138
      "source": [
139
        "while(True):\n",
140
        "    question = input(\"Enter question: \")\n",
141
        "    if question in ['q', 'exit', 'quit']:\n",
142
        "        break\n",
143
        "    answer = app.query(question)\n",
144
        "    print(answer)"
145
      ]
146
    }
147
  ],
148
  "metadata": {
149
    "colab": {
150
      "provenance": []
151
    },
152
    "kernelspec": {
153
      "display_name": "Python 3",
154
      "name": "python3"
155
    },
156
    "language_info": {
157
      "name": "python"
158
    }
159
  },
160
  "nbformat": 4,
161
  "nbformat_minor": 0
162
}
163