DeepImagix commited on
Commit
6b3320b
Β·
verified Β·
1 Parent(s): ac25004

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +9 -2
main.py CHANGED
@@ -61,7 +61,7 @@ from sklearn.linear_model import SGDClassifier
61
  from sklearn.pipeline import Pipeline
62
 
63
  from agent import agent_router
64
-
65
  try:
66
  from bs4 import BeautifulSoup
67
  BS4_AVAILABLE = True
@@ -273,7 +273,14 @@ async def lifespan(app: FastAPI):
273
  # ─────────────────────────────────────────────────────────────
274
  app = FastAPI(title="NeuraPrompt AI v7", lifespan=lifespan)
275
  app.include_router(studio_router)
276
- app.include_router(agent_router)
 
 
 
 
 
 
 
277
 
278
  cred = credentials.Certificate("serviceAccountKey.json")
279
  firebase_admin.initialize_app(cred)
 
61
  from sklearn.pipeline import Pipeline
62
 
63
  from agent import agent_router
64
+ from agent.tools.github_tools import github_router, register_github_tools
65
  try:
66
  from bs4 import BeautifulSoup
67
  BS4_AVAILABLE = True
 
273
  # ─────────────────────────────────────────────────────────────
274
  app = FastAPI(title="NeuraPrompt AI v7", lifespan=lifespan)
275
  app.include_router(studio_router)
276
+ app.include_router(agent_router
277
+
278
+ # Add router
279
+ app.include_router(github_router)
280
+
281
+ # Register tools
282
+ register_github_tools()
283
+
284
 
285
  cred = credentials.Certificate("serviceAccountKey.json")
286
  firebase_admin.initialize_app(cred)