sync with main

Cinnamon · Apr 1, 2024 · 4ab18c7 · 4ab18c7
1 parent 8e2f47d
commit 4ab18c7
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 11 deletions.
diff --git a/libs/kotaemon/kotaemon/indices/ingests/files.py b/libs/kotaemon/kotaemon/indices/ingests/files.py
@@ -8,7 +8,6 @@
 from kotaemon.indices.splitters import BaseSplitter, TokenSplitter
 from kotaemon.loaders import (
     AdobeReader,
-    AutoReader,
     DirectoryReader,
     MathpixPDFReader,
     OCRReader,

diff --git a/libs/kotaemon/kotaemon/loaders/adobe_loader.py b/libs/kotaemon/kotaemon/loaders/adobe_loader.py
@@ -41,7 +41,7 @@ class AdobeReader(BaseReader):
         ```
     Args:
         endpoint: URL to the Vision Language Model endpoint. If not provided,
-        will use the default `knowledgehub.loaders.adobe_loader.DEFAULT_VLM_ENDPOINT`
+        will use the default `kotaemon.loaders.adobe_loader.DEFAULT_VLM_ENDPOINT`
 
         max_figures_to_caption: an int decides how many figured will be captioned.
         The rest will be ignored (are indexed without captions).

diff --git a/libs/kotaemon/pyproject.toml b/libs/kotaemon/pyproject.toml
@@ -60,6 +60,7 @@ adv = [
     "cohere",
     "elasticsearch",
     "llama-cpp-python",
+    "pdfservices-sdk @  git+https://github.com/niallcm/pdfservices-python-sdk.git@bump-and-unfreeze-requirements",
 ]
 dev = [
     "ipython",

diff --git a/libs/ktem/ktem/reasoning/simple.py b/libs/ktem/ktem/reasoning/simple.py
@@ -165,15 +165,6 @@ def run(self, docs: list[RetrievedDocument]) -> Document:
     "Answer: "
 )
 
-DEFAULT_VLM_ENDPOINT = (
-    "{0}/openai/deployments/{1}/chat/completions?api-version={2}".format(
-        config("AZURE_OPENAI_ENDPOINT", default=""),
-        "gpt-4-vision",
-        config("OPENAI_API_VERSION", default=""),
-    )
-)
-
-
 class AnswerWithContextPipeline(BaseComponent):
     """Answer the question based on the evidence