chore(docs): added gemini reference

georgia-tech-db · Mar 7, 2024 · bc54da4 · bc54da4
1 parent a7aa70a
commit bc54da4
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 7 deletions.
diff --git a/docs/_toc.yml b/docs/_toc.yml
@@ -39,7 +39,7 @@ parts:
 
   - caption: User Reference
     chapters:
-      - file: source/reference/evaql      
+      - file: source/reference/evaql
         title: Query Language
         sections:
           - file: source/reference/evaql/load_csv
@@ -65,13 +65,13 @@ parts:
 
       - file: source/reference/api
         title: Python API
-      
+
       - file: source/reference/rest_api
         title: REST API
 
       - file: source/reference/databases/index
         title: Data Sources
-        sections: 
+        sections:
           - file: source/reference/databases/postgres
           - file: source/reference/databases/sqlite
           - file: source/reference/databases/mysql
@@ -83,7 +83,7 @@ parts:
 
       - file: source/reference/vector_databases/index
         title: Vector Databases
-        sections: 
+        sections:
           - file: source/reference/vector_databases/faiss
           - file: source/reference/vector_databases/chromadb
           - file: source/reference/vector_databases/qdrant
@@ -106,9 +106,11 @@ parts:
           - file: source/reference/ai/hf
             title: Hugging Face
           - file: source/reference/ai/openai
-            title: OpenAI 
+            title: OpenAI
+          - file: source/reference/ai/gemini
+            title: Gemini
           - file: source/reference/ai/yolo
-            title: YOLO 
+            title: YOLO
           - file: source/reference/ai/stablediffusion
             title: Stable Diffusion
 
@@ -117,7 +119,7 @@ parts:
 
       - file: source/reference/optimizations
         title: Optimizations
-        
+
       # - file: source/reference/io
       #   title: IO Descriptors
 

diff --git a/docs/source/reference/ai/gemini.rst b/docs/source/reference/ai/gemini.rst
@@ -0,0 +1,24 @@
+.. _gemini:
+
+Gemini Models
+=====================
+
+This section provides an overview of how you can use Gemini models in EvaDB.
+
+
+Chat Completion Functions
+-------------------------
+
+To create a chat completion function in EvaDB, use the following SQL command:
+
+.. code-block:: sql
+
+    CREATE FUNCTION IF NOT EXISTS GeminiChatCompletion
+    IMPL 'evadb/functions/gemini.py'
+    MODEL 'gemini-pro'
+
+EvaDB supports the following models for chat completion task:
+
+- "gemini-pro"
+
+The chat completion function can be composed in interesting ways with other functions. Gemini can be used similar to the `ChatGPT` function as shown in `Google Colab <https://colab.research.google.com/github/georgia-tech-db/evadb/blob/master/tutorials/08-chatgpt.ipynb>`_. as an example of combining chat completion task with caption extraction and video summarization models from Hugging Face and feeding it to chat completion to ask questions about the results.