petter2025 commited on
Commit
0491b8c
·
verified ·
1 Parent(s): 642b6b3

Update hf_demo.py

Browse files
Files changed (1) hide show
  1. hf_demo.py +4 -6
hf_demo.py CHANGED
@@ -1,9 +1,8 @@
1
- # hf_demo.py – ARF v4 API
2
  from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
4
  import gradio as gr
5
  from agentic_reliability_framework.core.governance.risk_engine import RiskEngine
6
- from agentic_reliability_framework.core.memory.semantic_memory import SemanticMemory
7
 
8
  app = FastAPI(title="ARF v4 API")
9
 
@@ -14,9 +13,8 @@ app.add_middleware(
14
  allow_methods=["*"],
15
  )
16
 
17
- # Initialize ARF components
18
  risk_engine = RiskEngine()
19
- memory = SemanticMemory()
20
 
21
  @app.get("/health")
22
  async def health():
@@ -24,14 +22,14 @@ async def health():
24
 
25
  @app.get("/api/v1/get_risk")
26
  async def get_risk():
27
- # Your existing risk endpoint logic
28
  risk_score = risk_engine.get_current_risk()
29
  return {
30
  "system_risk": risk_score.mean,
31
  "status": "critical" if risk_score.mean > 0.8 else "normal"
32
  }
33
 
34
- # Optional: keep the Gradio interface for interactive testing
35
  iface = gr.Interface(
36
  fn=lambda: f"ARF v4 - Current risk: {risk_engine.get_current_risk().mean:.2f}",
37
  inputs=[],
 
1
+ # hf_demo.py – ARF v4 API (Minimal)
2
  from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
4
  import gradio as gr
5
  from agentic_reliability_framework.core.governance.risk_engine import RiskEngine
 
6
 
7
  app = FastAPI(title="ARF v4 API")
8
 
 
13
  allow_methods=["*"],
14
  )
15
 
16
+ # Initialize RiskEngine only
17
  risk_engine = RiskEngine()
 
18
 
19
  @app.get("/health")
20
  async def health():
 
22
 
23
  @app.get("/api/v1/get_risk")
24
  async def get_risk():
25
+ # Get current system risk (this method exists in v4)
26
  risk_score = risk_engine.get_current_risk()
27
  return {
28
  "system_risk": risk_score.mean,
29
  "status": "critical" if risk_score.mean > 0.8 else "normal"
30
  }
31
 
32
+ # Optional Gradio interface
33
  iface = gr.Interface(
34
  fn=lambda: f"ARF v4 - Current risk: {risk_engine.get_current_risk().mean:.2f}",
35
  inputs=[],