backend/scripts/test_krs_graph.py

import os
import sys

# Ustaw ścieżkę do backendu
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))

from dotenv import load_dotenv

if not load_dotenv(os.path.join(os.path.dirname(__file__), "..", "..", ".env")):
    print("Nie udało się pobrać .env")

from agents.tools.krs_graph_tool import analyze_company_network


def main():
    # Sprawdźmy dowolny KRS np PKN Orlen, CD Projekt, PKO BP, ale lepiej wziąć dużą znaną spółkę,
    # np CD Projekt ma KRS: 0000006865, NIP: 7342867148
    # ALbo inna - Allegro? Żabka?
    # Dla testu użyjemy CD Projekt: 0000006865
    test_krs = "0000006865"

    print(f"Pobieranie odpisu z KRS dla {test_krs}...")
    result = analyze_company_network.invoke({"krs_number": test_krs})

    import json

    print("\n--- ZWROCONE DANE DO AGENTA DO ANALIZY MŚP ---")
    print(json.dumps(result, indent=2, ensure_ascii=False))


if __name__ == "__main__":
    main()