updating for latest embedding models (#1129)

This commit is contained in:
Joe Beutler 2024-04-01 17:04:55 -07:00 committed by GitHub
parent 7c3aaa85c8
commit a405468580
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -31,7 +31,7 @@
"client = OpenAI()\n", "client = OpenAI()\n",
"\n", "\n",
"embedding = client.embeddings.create(\n", "embedding = client.embeddings.create(\n",
" input=\"Your text goes here\", model=\"text-embedding-ada-002\"\n", " input=\"Your text goes here\", model=\"text-embedding-3-small\"\n",
").data[0].embedding\n", ").data[0].embedding\n",
"len(embedding)\n" "len(embedding)\n"
] ]
@ -57,7 +57,7 @@
"num_embeddings = 10000 # Some large number\n", "num_embeddings = 10000 # Some large number\n",
"for i in range(num_embeddings):\n", "for i in range(num_embeddings):\n",
" embedding = client.embeddings.create(\n", " embedding = client.embeddings.create(\n",
" input=\"Your text goes here\", model=\"text-embedding-ada-002\"\n", " input=\"Your text goes here\", model=\"text-embedding-3-small\"\n",
" ).data[0].embedding\n", " ).data[0].embedding\n",
" print(len(embedding))" " print(len(embedding))"
] ]