File size: 630 Bytes
7690851 de597ec 7690851 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import os
import pandas as pd
import numpy as np
from dotenv import load_dotenv
from ml.inference import MLEngine
from agent.agent import MaintenanceAgent
load_dotenv()
raw_df = pd.DataFrame({
"vdc1": np.random.normal(600, 3, 200),
"idc1": np.random.normal(10.0, 0.2, 200)
})
engine = MLEngine()
phase2_output = engine.predict_from_raw(raw_df)
print("\n=== ML OUTPUT ===")
print(phase2_output)
agent = MaintenanceAgent(
api_key=os.getenv("GOOGLE_API_KEY"),
model_name="gemini-2.5-flash-lite",
temperature=0.0
)
agent_output = agent.run(phase2_output)
print("\n=== AGENT OUTPUT ===")
print(agent_output)
|