Spaces:
Running
Running
Update main.py
Browse files
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)
|