Ed_
60396f03f8
Applied 236 return type annotations to functions with no return values across 100+ files (core modules, tests, scripts, simulations). Added Phase 4 to python_style_refactor track for remaining 597 items (untyped params, vars, and functions with return values). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
36 lines
810 B
Python
36 lines
810 B
Python
# gemini.py
|
|
import tomllib
|
|
from pathlib import Path
|
|
from google import genai
|
|
from google.genai import types
|
|
|
|
_client = None
|
|
_chat = None
|
|
|
|
def _load_key() -> str:
|
|
with open("credentials.toml", "rb") as f:
|
|
return tomllib.load(f)["gemini"]["api_key"]
|
|
|
|
def _ensure_client() -> None:
|
|
global _client
|
|
if _client is None:
|
|
_client = genai.Client(api_key=_load_key())
|
|
|
|
def _ensure_chat() -> None:
|
|
global _chat
|
|
if _chat is None:
|
|
_ensure_client()
|
|
_chat = _client.chats.create(model="gemini-2.0-flash")
|
|
|
|
def send(md_content: str, user_message: str) -> str:
|
|
global _chat
|
|
_ensure_chat()
|
|
full_message = f"<context>\n{md_content}\n</context>\n\n{user_message}"
|
|
response = _chat.send_message(full_message)
|
|
return response.text
|
|
|
|
def reset_session() -> None:
|
|
global _client, _chat
|
|
_client = None
|
|
_chat = None
|