Skip to content

Cleanup: Centralize transaction categories (Issue #24) #45

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
dhabyap merged 2 commits into from
Apr 14, 2026
Merged

Cleanup: Centralize transaction categories (Issue #24) #45

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
9197ff9
Centralize transaction categories to config.py (Closes #24)
dhabyap Apr 14, 2026
fdecbfb
Merge branch 'main' into cleanup/centralize-categories-issue-24
dhabyap Apr 14, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 12 additions & 8 deletions ai_brain.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,27 +5,31 @@
from google.genai import types
from openai import OpenAI
from dotenv import load_dotenv
import config

load_dotenv()

# System instruction yang padat agar model AI lebih cepat merespons
SYSTEM_INSTRUCTION = """
def get_system_instruction():
categories_str = " | ".join([f'"{cat}"' for cat in config.TRANSACTION_CATEGORIES])
instruction = f"""
Kamu adalah asisten AI spesialis pencatatan keuangan pribadi. Tugasmu adalah mengekstrak data dari pesan pengguna ke format JSON murni.

SKEMA JSON:
{
{{
"tipe": "pemasukan" | "pengeluaran" | "investasi",
"item": "deskripsi singkat transaksi",
"nominal": angka integer,
"kategori": "makanan_minuman" | "transportasi" | "tagihan_rutin" | "hiburan" | "belanja_pribadi" | "pendapatan_gaji" | "pendapatan_sampingan" | "aset_investasi" | "lainnya"
}
"kategori": {categories_str}
}}

ATURAN KETAT:
1. Konversi singkatan angka: "k" / "rb" = 1000, "jt" / "juta" = 1000000.
2. Jika nominal dalam teks (misal: "setengah juta", "seratus ribu"), ubah ke angka integer (misal: 500000, 100000).
3. Fokus pada uang masuk, keluar, atau investasi.
4. Jawaban HANYA berupa JSON murni tanpa teks pengantar. Jika error, kembalikan {"error": true}.
4. Jawaban HANYA berupa JSON murni tanpa teks pengantar. Jika error, kembalikan {{"error": true}}.
"""
return instruction

def call_gemini(text: str) -> str:
"""Memanggil Google Gemini API."""
Expand All @@ -38,7 +42,7 @@ def call_gemini(text: str) -> str:
model='gemini-2.0-flash',
contents=text,
config=types.GenerateContentConfig(
system_instruction=SYSTEM_INSTRUCTION,
system_instruction=get_system_instruction(),
response_mime_type="application/json",
temperature=0.1
)
Expand All @@ -55,7 +59,7 @@ def call_openai_compatible(text: str, api_key: str, base_url: str, model: str) -
response = client.chat.completions.create(
model=model,
messages=[
{"role": "system", "content": SYSTEM_INSTRUCTION},
{"role": "system", "content": get_system_instruction()},
{"role": "user", "content": text}
],
temperature=0.1
Expand All @@ -77,7 +81,7 @@ def call_cohere(text: str) -> str:
json={
"model": "command-r",
"message": text,
"preamble": SYSTEM_INSTRUCTION,
"preamble": get_system_instruction(),
"temperature": 0.1
}
)
Expand Down
5 changes: 5 additions & 0 deletions config.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,3 +19,8 @@
5: "Mei", 6: "Juni", 7: "Juli", 8: "Agustus",
9: "September", 10: "Oktober", 11: "November", 12: "Desember"
}
TRANSACTION_CATEGORIES = [
"makanan_minuman", "transportasi", "tagihan_rutin", "hiburan",
"belanja_pribadi", "pendapatan_gaji", "pendapatan_sampingan",
"aset_investasi", "lainnya"
]
Loading