feat: bump llama index after pr was accepted

lavague-ai · May 16, 2024 · 0c86a92 · 0c86a92
1 parent 7605e17
commit 0c86a92
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 248 deletions.
diff --git a/lavague-integrations/contexts/lavague-contexts-gemini/lavague/contexts/gemini/base.py b/lavague-integrations/contexts/lavague-contexts-gemini/lavague/contexts/gemini/base.py
@@ -12,7 +12,6 @@
     Context,
     get_default_context,
 )
-from .base_mml import GeminiMultiModal2
 from lavague.core.extractors import BaseExtractor
 from lavague.core.retrievers import BaseHtmlRetriever
 from lavague.core.context import DEFAULT_MAX_TOKENS, DEFAULT_TEMPERATURE
@@ -40,7 +39,7 @@ def __init__(
                 max_tokens=DEFAULT_MAX_TOKENS,
                 temperature=DEFAULT_TEMPERATURE,
             ),
-            GeminiMultiModal2(api_key=api_key, model_name=mm_llm),
+            GeminiMultiModal(api_key=api_key, model_name=mm_llm),
             GeminiEmbedding(api_key=api_key, model_name=embedding),
             retriever,
             prompt_template,

diff --git a/lavague-integrations/contexts/lavague-contexts-gemini/lavague/contexts/gemini/base_mml.py b/lavague-integrations/contexts/lavague-contexts-gemini/lavague/contexts/gemini/base_mml.py
diff --git a/lavague-integrations/contexts/lavague-contexts-gemini/pyproject.toml b/lavague-integrations/contexts/lavague-contexts-gemini/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lavague-contexts-gemini"
-version = "0.1.2"
+version = "0.1.3"
 description = "gemini integration for lavague"
 authors = ["lavague-ai"]
 readme = "README.md"
@@ -22,9 +22,9 @@ packages = [{include = "lavague/"}]
 [tool.poetry.dependencies]
 python = "^3.10.0"
 lavague-core = "^0.1.1"
-llama-index-llms-gemini = "^0.1.7"
+llama-index-llms-gemini = "^0.1.8"
 llama-index-embeddings-gemini = "^0.1.6"
-llama-index-multi-modal-llms-gemini = "^0.1.5"
+llama-index-multi-modal-llms-gemini = "^0.1.6"
 google-generativeai = "^0.4.1"
 
 [build-system]