added encode_gemini(text,key)
Browse files- library.ipynb +16 -0
library.ipynb
CHANGED
|
@@ -55,6 +55,22 @@
|
|
| 55 |
" return -1,str(e)\n",
|
| 56 |
" return 0,prediction"
|
| 57 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
}
|
| 59 |
],
|
| 60 |
"metadata": {
|
|
|
|
| 55 |
" return -1,str(e)\n",
|
| 56 |
" return 0,prediction"
|
| 57 |
]
|
| 58 |
+
},
|
| 59 |
+
{
|
| 60 |
+
"cell_type": "code",
|
| 61 |
+
"execution_count": null,
|
| 62 |
+
"metadata": {},
|
| 63 |
+
"outputs": [],
|
| 64 |
+
"source": [
|
| 65 |
+
"def encode_gemini(text,key):\n",
|
| 66 |
+
" genai.configure(api_key=key)\n",
|
| 67 |
+
" result = genai.embed_content(\n",
|
| 68 |
+
" model=\"models/embedding-001\",\n",
|
| 69 |
+
" content=text,\n",
|
| 70 |
+
" task_type=\"retrieval_document\",\n",
|
| 71 |
+
" title=\"Embedding of single string\")\n",
|
| 72 |
+
" return result['embedding']"
|
| 73 |
+
]
|
| 74 |
}
|
| 75 |
],
|
| 76 |
"metadata": {
|