abstra 3.24.0__py3-none-any.whl → 3.24.1__py3-none-any.whl

abstra_internals/controllers/ai.py

@@ -1,10 +1,17 @@
+import base64
+import io
 from dataclasses import dataclass
 from typing import Any, Dict, List
 
+import pypdfium2 as pdfium
+
 from abstra_internals.cloud_api import get_session_path, get_tunnel_secret_key
 from abstra_internals.contracts_generated import (
     AbstraLibApiAiStreamRequest,
     CloudApiCliAiV2StreamRequest,
+    CloudApiCliAiV2StreamRequestContentItem,
+    CloudApiCliAiV2StreamRequestContentItemAssistantFileInput,
+    CloudApiCliAiV2StreamRequestContentItemAssistantTextInput,
 )
 from abstra_internals.controllers.main import MainController
 from abstra_internals.credentials import resolve_headers
@@ -71,15 +78,96 @@ class AiController:
         self.controller = controller
         self.repos = controller.repositories
 
+    def _extract_pdf_images(self, pdf_bytes: bytes) -> List[bytes]:
+        images = []
+        pdf_io = io.BytesIO(pdf_bytes)
+        for page in pdfium.PdfDocument(pdf_io):
+            bitmap = page.render(
+                scale=4,  # 288 dpi
+                rotation=0,
+            )
+            pil_image = bitmap.to_pil()
+            image_io = io.BytesIO()
+            pil_image.save(image_io, format="png")
+            image_io.seek(0)
+            images.append(image_io.read())
+        return images
+
+    def _process_pdf_content(
+        self, pdf_file: CloudApiCliAiV2StreamRequestContentItemAssistantFileInput
+    ) -> List[CloudApiCliAiV2StreamRequestContentItem]:
+        processed_items = []
+        try:
+            pdf_bytes = base64.b64decode(pdf_file.file_content)
+            images = self._extract_pdf_images(pdf_bytes)
+
+            processed_items.append(
+                CloudApiCliAiV2StreamRequestContentItemAssistantTextInput(
+                    type="text",
+                    text=f"[PDF Document: {pdf_file.file_name} - {len(images)} page(s)]",
+                )
+            )
+
+            for idx, image_bytes in enumerate(images, 1):
+                image_base64 = base64.b64encode(image_bytes).decode("utf-8")
+                processed_items.append(
+                    CloudApiCliAiV2StreamRequestContentItemAssistantFileInput(
+                        type="file",
+                        file_name=f"{pdf_file.file_name}_page_{idx}.png",
+                        file_content=image_base64,
+                    )
+                )
+        except Exception as e:
+            print(f"Error extracting PDF images: {e}")
+            processed_items.append(
+                CloudApiCliAiV2StreamRequestContentItemAssistantTextInput(
+                    type="text",
+                    text=f"[PDF Document: {pdf_file.file_name}]\nNote: Could not extract images from this PDF.",
+                )
+            )
+        return processed_items
+
+    def _process_content(
+        self, content: List[CloudApiCliAiV2StreamRequestContentItem]
+    ) -> List[CloudApiCliAiV2StreamRequestContentItem]:
+        processed_content = []
+
+        for item in content:
+            if isinstance(
+                item, CloudApiCliAiV2StreamRequestContentItemAssistantTextInput
+            ):
+                processed_content.append(item)
+                continue
+
+            if not item.file_name.lower().endswith(".pdf"):
+                processed_content.append(item)
+                continue
+
+            try:
+                processed_pdf_items = self._process_pdf_content(item)
+                processed_content.extend(processed_pdf_items)
+            except Exception as e:
+                print(f"Error extracting PDF images: {e}")
+                processed_content.append(
+                    CloudApiCliAiV2StreamRequestContentItemAssistantTextInput(
+                        type="text",
+                        text=f"[PDF Document: {item.file_name}]\nNote: Could not extract images from this PDF.",
+                    )
+                )
+
+        return processed_content
+
     def send_ai_message(
         self,
         body: AbstraLibApiAiStreamRequest,
     ):
         try:
+            processed_content = self._process_content(body.content)
+
             yield from self.repos.ai.get_ai_messages(
                 CloudApiCliAiV2StreamRequest(
                     conversation_id=body.conversation_id,
-                    content=body.content,
+                    content=processed_content,
                     context={
                         **body.context,
                         "libVersion": str(get_local_package_version()),

abstra_internals/controllers/git.py

@@ -165,6 +165,8 @@ class GitController:
 
         status_response = GitStatusResponse(
             available=status.available,
+            git_installed=status.git_installed,
+            is_monorepo=status.is_monorepo,
             branch=status.branch,
             branches=status.branches,
             last_commit=status.last_commit,

abstra_internals/interface/cli/editor.py

@@ -80,7 +80,7 @@ def ensure_certificates():
             print(f"Failed to restore certificates: {update_e}")
 
 
-def editor(headless: bool):
+def editor(headless: bool, verbose: bool = False):
     ensure_certificates()
 
     load_dotenv(Settings.root_path / ".env")
@@ -119,6 +119,6 @@ def editor(headless: bool):
     if not headless:
         background_open_editor()
 
-    connect_tunnel()
+    connect_tunnel(verbose=verbose)
 
     server.serve_forever()

abstra_internals/repositories/git/__init__.py

@@ -1,8 +1,3 @@
-import shutil
-
-from abstra_internals.environment import FORCE_GIT_CLIENT
-
-from .dulwich import DulwichGitRepository
 from .native import NativeGitRepository
 from .types import (
     AheadBehindInfo,
@@ -15,14 +10,7 @@ from .types import (
 
 
 def create_git_repository(working_directory) -> GitRepositoryInterface:
-    if FORCE_GIT_CLIENT == "native":
-        return NativeGitRepository(working_directory)
-    elif FORCE_GIT_CLIENT == "dulwich":
-        return DulwichGitRepository(working_directory)
-    elif shutil.which("git") is not None:
-        return NativeGitRepository(working_directory)
-    else:
-        return DulwichGitRepository(working_directory)
+    return NativeGitRepository(working_directory)
 
 
 __all__ = [
@@ -32,7 +20,6 @@ __all__ = [
     "GitStatusResponse",
     "GitRepositoryInterface",
     "NativeGitRepository",
-    "DulwichGitRepository",
     "create_git_repository",
     "AheadBehindInfo",
 ]