sec-gemini 0.0.1__tar.gz

sec_gemini-0.0.1/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: sec-gemini
+Version: 0.0.1
+Summary: Sec-Gemini Python SDK
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: asyncio>=3.4.3
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: openai>=1.59.7
+Requires-Dist: orjson>=3.10.12
+Requires-Dist: pydantic>=2.10.4
+Requires-Dist: python-dotenv>=1.0.1
+Requires-Dist: rich>=13.9.4
+Requires-Dist: websockets>=14.1
+Requires-Dist: dotenv
+
+# Sec-Gemini Python SDK
+

sec_gemini-0.0.1/README.md

@@ -0,0 +1,2 @@
+# Sec-Gemini Python SDK
+

sec_gemini-0.0.1/pyproject.toml

@@ -0,0 +1,27 @@
+[project]
+name = "sec-gemini"
+version = "0.0.1"
+description = "Sec-Gemini Python SDK"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "asyncio>=3.4.3",
+    "httpx>=0.28.1",
+    "openai>=1.59.7",
+    "orjson>=3.10.12",
+    "pydantic>=2.10.4",
+    "python-dotenv>=1.0.1",
+    "rich>=13.9.4",
+    "websockets>=14.1",
+    "dotenv" # what version?
+]
+
+[tool.uv]
+dev-dependencies = [
+    "ipykernel>=6.29.5",
+    "jupyter>=1.1.1",
+    "pytest>=8.3.4",
+]
+
+[tool.setuptools.packages.find]
+include = ["sec_gemini*"]

sec_gemini-0.0.1/sec_gemini/__init__.py

@@ -0,0 +1,34 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from .secgemini import SecGemini
+from .models.session_request import SessionRequest
+from .models.session_response import SessionResponse
+from .models.message import Message
+from .models.enums import State
+from .file import File
+from .session import InteractiveSession
+from .models.enums import MimeType, MessageType
+
+__all__ = [
+    "SecGemini",
+    "SessionRequest",
+    "SessionResponse",
+    "Message",
+    "File",
+    "InteractiveSession",
+    "MimeType",
+    "MessageType",
+    "State",
+]

sec_gemini-0.0.1/sec_gemini/constants.py

@@ -0,0 +1,15 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+DEFAULT_TTL = 86400 * 3 # 3 days

sec_gemini-0.0.1/sec_gemini/enums.py

@@ -0,0 +1,46 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from enum import Enum
+
+class SDKInfo(Enum):
+    """SDK Info"""
+    NAME = "secgemini-python"
+    VERSION = "0.2.0"  # reported in the header of the API
+
+class _URLS(Enum):
+    """API URLs"""
+    HTTPS = "https://api.secgemini.google"
+    WEBSOCKET = "wss://api.secgemini.google"
+
+class _EndPoints(Enum):
+    """API Endpoints"""
+    # users
+    USER_INFO = "/v1/user/info"
+
+    # messages
+    GENERATE = "/v1/session/generate"
+    STREAM = "/v1/stream"
+
+    # sessions
+    REGISTER_SESSION = "/v1/session/register"
+    DELETE_SESSION = "/v1/session/delete"
+    LIST_SESSION = "/v1/session/list"
+    GET_SESSION = "/v1/session/get"
+    UPDATE_SESSION = "/v1/session/update"
+    SEND_FEEDBACK = "/v1/session/feedback"
+
+    # files
+    ATTACH_FILE = "/v1/session/attach_file"
+    DELETE_FILE = "/v1/session/delete_file"

sec_gemini-0.0.1/sec_gemini/file.py

@@ -0,0 +1,39 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from pydantic import BaseModel, Field
+from .models.enums import MimeType
+
+class File(BaseModel):
+    """
+    Represents a file that can be uploaded to the API.
+    """
+
+    filename: str = Field(
+        ...,
+        title="Filename",
+        description="The name of the file."
+    )
+
+    mime_type: MimeType = Field(
+        ...,
+        title="Mime Type",
+        description="The mime type of the file."
+    )
+
+    file: bytes = Field(
+        ...,
+        title="File",
+        description="The file content."
+    )

sec_gemini-0.0.1/sec_gemini/http.py

@@ -0,0 +1,121 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import sys
+from time import time
+from pydantic import BaseModel, Field
+import httpx
+import logging
+import websockets
+from .enums import SDKInfo
+from typing import TypeVar
+# Define a TypeVar for subclasses of HTTPResponse
+T = TypeVar('T', bound='BaseModel')
+
+
+class NetResponse(BaseModel):
+    url: str = Field(title="Request URL")
+    ok: bool
+    error_message: str = Field("", title="Error Message")
+    data: dict = Field({}, title="Response Data")
+    latency: float = Field(0.0, title="Latency",
+                           description="The time taken to complete the request in seconds.")
+
+class NetworkClient():
+    def __init__(self,
+                 base_url: str,
+                 api_key: str):
+        self.base_url = base_url
+        self.api_key = api_key
+        self.client = httpx.Client(timeout=90)
+
+
+    def post(self, endpoint: str, model: T, headers: dict = {}) -> NetResponse:
+        """Post Request to the API
+
+        Args:
+            endpoint: The API endpoint to post to.
+            model: The pydantic model to be posted.
+            headers: The headers specific to the requests to be sent along.
+
+        Returns:
+            HTTPResponse: The response from the API.
+        """
+        data = model.model_dump()
+        url = self._make_url(endpoint)
+        headers = self._make_headers(headers)
+
+        start_time = time()
+        response = self.client.post(url, headers=headers, json=data)
+        latency = time() - start_time
+
+        if response.status_code != 200:
+            return NetResponse(url=url,
+                                ok=False,
+                                error_message=self._make_error_message(url, response),
+                                latency=latency)
+
+        logging.debug(f"[HTTP][POST] {url} -> latency: {latency}")
+        return NetResponse(url=url, ok=True, data=response.json(),
+                            latency=latency)
+
+    def get(self, endpoint: str, query_params: dict = {}, headers: dict = {}) -> NetResponse:
+        """Get Request to the API
+
+        Args:
+            endpoint: The API endpoint to get from.
+            query_params: The query parameters to be sent along.
+            headers: The headers specific to the requests to be sent along.
+
+        Returns:
+            HTTPResponse: The response from the API.
+        """
+
+        # FIXME: Support for query parameters
+
+        url = self._make_url(endpoint)
+        headers = self._make_headers(headers)
+        start_time = time()
+        response = self.client.get(url, params=query_params, headers=headers)
+        latency = time() - start_time
+        if response.status_code != 200:
+            return NetResponse(url=url,
+                                ok=False,
+                                error_message=self._make_error_message(url, response),
+                                latency=latency)
+        logging.debug(f"[HTTP][GET] {url} -> latency: {latency}")
+        return NetResponse(url=url, ok=True, data=response.json(),
+                            latency=latency)
+
+    def _make_url(self, endpoint: str) -> str:
+        return f"{self.base_url}/{endpoint.lstrip('/')}"
+
+    def _make_headers(self, headers: dict) -> dict:
+        # User-Agent: Mozilla/5.0 (<system-information>) <platform> (<platform-details>) <extensions>
+
+        # request specific headers
+        headers = headers or {}
+
+        additional_headers = {
+            "User-Agent": f"{SDKInfo.NAME.value}/{SDKInfo.VERSION.value} ({sys.platform}) {sys.version} ({sys.version_info})",
+            "x-sdk-version": SDKInfo.VERSION.value,
+            "x-sdk": "python",
+            "x-api-key": self.api_key,
+            "Content-Type": "application/json"
+        }
+        headers.update(additional_headers)
+        return headers
+
+    def _make_error_message(self, url: str, response: httpx.Response) -> str:
+        return f"[HTTP] {url} -> {response.status_code}:{response.text}"

sec_gemini-0.0.1/sec_gemini/models/attachment.py

@@ -0,0 +1,37 @@
+
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from pydantic import BaseModel, Field
+from .enums import MimeType
+
+class Attachment(BaseModel):
+    """Represents a file upload to the session."""
+
+    session_id: str = Field(...,
+                            title="Session ID",
+                            description="The session ID this file should be attached to.")
+
+    filename: str = Field(...,
+                          title="Filename",
+                          description="The name of the file.")
+
+    mime_type: MimeType = Field(...,
+                                title="Mime Type",
+                                description="The mime type of the file.")
+
+    content: str = Field(...,
+                           title="File Content",
+                           description="The content of the file as string. Always base64 encoded.")

sec_gemini-0.0.1/sec_gemini/models/enums.py

@@ -0,0 +1,222 @@
+
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from enum import Enum
+
+class FeedbackType(str, Enum):
+    """Type of feedback that can be sent to the system."""
+
+    USER_FEEDBACK = "user_feedback"
+    BUG_REPORT = "bug_report"
+
+class VectorDistance(str, Enum):
+    COSINE = "COSINE"
+    DOT_PRODUCT = "DOT_PRODUCT"
+    EUCLIDEAN = "EUCLIDEAN"
+
+class UserType(str, Enum):
+    "User type"
+    UI = "ui"   # user interface
+    USER = "user"  # system user
+    ADMIN = "admin"  # admin user
+    SYSTEM = "system"  # orchestrator system user
+    SERVICE = "service" # microservice
+
+class ResponseStatus(int, Enum):
+    # 2xx Success
+    OK = 200
+    CREATED = 201
+    ACCEPTED = 202
+    NO_CONTENT = 204
+    PARTIAL_CONTENT = 206
+
+    # 3xx Redirection
+    MULTIPLE_CHOICES = 300
+    MOVED_PERMANENTLY = 301
+    FOUND = 302
+    SEE_OTHER = 303
+    NOT_MODIFIED = 304
+    TEMPORARY_REDIRECT = 307
+    PERMANENT_REDIRECT = 308
+
+    # 4xx Client Errors
+    BAD_REQUEST = 400
+    UNAUTHORIZED = 401
+    AUTHENTICATION_ERROR = 401  # Alias for UNAUTHORIZED
+    PAYMENT_REQUIRED = 402
+    FORBIDDEN = 403
+    NOT_FOUND = 404
+    METHOD_NOT_ALLOWED = 405
+    NOT_ACCEPTABLE = 406
+    PROXY_AUTHENTICATION_REQUIRED = 407
+    REQUEST_TIMEOUT = 408
+    CONFLICT = 409
+    ALREADY_EXISTS = 409  # Alias for CONFLICT
+    GONE = 410
+    LENGTH_REQUIRED = 411
+    PRECONDITION_FAILED = 412
+    PAYLOAD_TOO_LARGE = 413
+    URI_TOO_LONG = 414
+    UNSUPPORTED_MEDIA_TYPE = 415
+    RANGE_NOT_SATISFIABLE = 416
+    EXPECTATION_FAILED = 417
+    I_AM_A_TEAPOT = 418
+    UNPROCESSABLE_ENTITY = 422
+    TOO_EARLY = 425
+    UPGRADE_REQUIRED = 426
+    PRECONDITION_REQUIRED = 428
+    TOO_MANY_REQUESTS = 429
+    QUOTA_EXCEEDED = 429  # Alias for TOO_MANY_REQUESTS
+    REQUEST_HEADER_FIELDS_TOO_LARGE = 431
+    UNAVAILABLE_FOR_LEGAL_REASONS = 451
+
+    # 5xx Server Errors
+    INTERNAL_SERVER_ERROR = 500
+    SERVER_ERROR = 500  # Alias for INTERNAL_SERVER_ERROR
+    INTERNAL_ERROR = 500  # Another alias for INTERNAL_SERVER_ERROR
+    NOT_IMPLEMENTED = 501
+    BAD_GATEWAY = 502
+    SERVICE_UNAVAILABLE = 503
+    GATEWAY_TIMEOUT = 504
+    HTTP_VERSION_NOT_SUPPORTED = 505
+    VARIANT_ALSO_NEGOTIATES = 506
+    INSUFFICIENT_STORAGE = 507
+    LOOP_DETECTED = 508
+    NOT_EXTENDED = 510
+    NETWORK_AUTHENTICATION_REQUIRED = 511
+
+class Role(str, Enum):
+    "Describe the role associated with the completion"
+    USER = "user"
+    AGENT = "agent"
+    SYSTEM = "system"   # those are not returned to the user
+
+class MimeType(str, Enum):
+    "Completion type"
+    TEXT = "text/plain"
+    MARKDOWN = "text/markdown"
+    SERIALIZED_JSON = "text/serialized-json"
+    BINARY = "application/octet-stream"
+
+    # IMAGES
+    JPEG = "image/jpeg"
+    PNG = "image/png"
+    TIFF = "image/tiff"
+    GIF = "image/gif"
+    SVG = "image/svg+xml"
+    WEBP = "image/webp"
+    AVIF = "image/avif"
+
+    # AUDIO
+    WAV = "audio/wav"
+    MP3 = "audio/mpeg"
+    OGG = "audio/ogg"
+
+    # VIDEO
+    WEBM = "video/webm"
+    MP4 = "video/mp4"
+
+
+    # CODE
+    C = "text/c"
+    CPP = "text/c++"
+    JAVA = "text/java"
+    RUST = "text/rust"
+    GOLANG = "text/go"
+    PYTHON = "text/python"
+    PHP = "text/php"
+    PERL = "text/perl"
+    RUBY = "text/ruby"
+    SWIFT = "text/swift"
+    KOTLIN = "text/kotlin"
+    SCALA = "text/scala"
+    JAVASCRIPT = "text/javascript"
+    TYPESCRIPT = "text/typescript"
+    HTML = "text/html"
+    CSS = "text/css"
+
+    # DATA
+    CSV = "text/csv"
+    XML = "text/xml"
+    YAML = "text/yaml"
+    TOML = "text/toml"
+    SQL = "text/sql"
+    JSON = "application/json"
+    JSONL = "application/jsonl"
+
+
+    # COMPRESSED
+    # NOTE: Gemini does not support compressed files
+    # ZIP = "application/zip"
+    # TAR = "application/tar"
+    # GZIP = "application/gzip"
+    # BZIP2 = "application/bzip2"
+    # XZ = "application/xz"
+    # SEVENZIP = "application/x-7z-compressed"
+
+    # DOCUMENTS
+    PDF = "application/pdf"
+    DOCX = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+    XLSX = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+    PPTX = "application/vnd.openxmlformats-officedocument.presentationml.presentation"
+    DOC = "application/msword"
+    XLS = "application/vnd.ms-excel"
+    PPT = "application/vnd.ms-powerpoint"
+    RTF = "application/rtf"
+    ODT = "application/vnd.oasis.opendocument.text"
+
+
+class State(str, Enum):
+    START  = "start"  # session just started
+
+    QUERY = "query"  # user query
+
+    RUNNING_AGENT = "running_agent"  # executing agent
+    AGENT_DONE = "agent_done"  # agent done
+
+    CODING = "coding"  # executing code
+    CODE_RESULT = "code_result"  # code result
+
+    CALLING_TOOL = "calling_tool"  # executing function
+    TOOL_RESULT = "tool_result"  # function result
+
+    # semantic sugar states
+    GENERATING = "generating"  # generating response
+    ANSWERING = "answering"  # generating answer
+    THINKING = "thinking"  # thinking
+    PLANNING = "planning"  # planning execution
+    REVIEWING = "reviewing"  # reviewing current result
+    UNDERSTANDING = "understanding"  # intent detection
+    RETRIVING = "retriving"  # retrieving info
+    GROUNDING = "grounding"  # grounding
+
+
+class MessageType(str, Enum):
+    "Type of message"
+
+    # info messages
+    RESULT = "result"  # result message
+    DEBUG = "debug"  # debug message
+    INFO = "info"   # transient info message only used in streaming
+    ERROR = "error"  # error message
+    THINKING = "thinking"  # thinking message that persist in the thinking panel
+
+    # mutation messages
+    UPDATE = "update"  # update message that modify the output. e.g grounding or new fact
+    DELETE = "delete"  # Ask to delete a previous message by id
+
+    # User messages
+    QUERY = "query"

sec_gemini-0.0.1/sec_gemini/models/feedback.py

@@ -0,0 +1,37 @@
+
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from pydantic import BaseModel, Field
+from .enums import FeedbackType
+
+
+class Feedback(BaseModel):
+    """Represents a feedback to the session."""
+
+    session_id: str = Field(..., title="Session ID",
+                            description="The session ID this feedback should be attached to.")
+
+    group_id: str = Field('', title="Group ID",
+                          description="The message group ID this feedback should be attached to.")
+
+    type: FeedbackType = Field(..., title="Feedback Type",
+                                         description="The type of feedback.")
+
+    score: int = Field(..., title="Score",
+                       description="The score of the feedback.")
+
+    comment: str = Field(..., title="Comment",
+                         description="The comment of the feedback.")