Spaces:
Sleeping
Sleeping
Lucas ARRIESSE
commited on
Commit
·
c1e5d8a
1
Parent(s):
8cc6fe4
Add doc
Browse files- app.py +2 -2
- docs/docs.md +9 -0
app.py
CHANGED
@@ -43,7 +43,7 @@ llm_router = Router(model_list=[
|
|
43 |
prompt_env = Environment(loader=FileSystemLoader(
|
44 |
'prompts'), enable_async=True, undefined=StrictUndefined)
|
45 |
|
46 |
-
api = FastAPI(docs_url="/")
|
47 |
# requirements routes
|
48 |
requirements_router = APIRouter(prefix="/reqs", tags=["requirements"])
|
49 |
# solution routes
|
@@ -114,7 +114,7 @@ async def categorize_reqs(params: ReqGroupingRequest) -> ReqGroupingResponse:
|
|
114 |
|
115 |
@solution_router.post("/search_solutions_gemini", response_model=SolutionSearchResponse)
|
116 |
async def search_solutions(params: ReqGroupingResponse) -> SolutionSearchResponse:
|
117 |
-
"""Searches solutions using Gemini and grounded on google search"""
|
118 |
|
119 |
async def _search_inner(cat: ReqGroupingCategory) -> SolutionModel:
|
120 |
# ================== generate the solution with web grounding
|
|
|
43 |
prompt_env = Environment(loader=FileSystemLoader(
|
44 |
'prompts'), enable_async=True, undefined=StrictUndefined)
|
45 |
|
46 |
+
api = FastAPI(docs_url="/", title="Reqxtract-API", description=open("docs/docs.md").read())
|
47 |
# requirements routes
|
48 |
requirements_router = APIRouter(prefix="/reqs", tags=["requirements"])
|
49 |
# solution routes
|
|
|
114 |
|
115 |
@solution_router.post("/search_solutions_gemini", response_model=SolutionSearchResponse)
|
116 |
async def search_solutions(params: ReqGroupingResponse) -> SolutionSearchResponse:
|
117 |
+
"""Searches solutions solving the given grouping params using Gemini and grounded on google search"""
|
118 |
|
119 |
async def _search_inner(cat: ReqGroupingCategory) -> SolutionModel:
|
120 |
# ================== generate the solution with web grounding
|
docs/docs.md
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Reqxtract-API
|
2 |
+
|
3 |
+
This project exposes the API used by reqxtract to perform the following functions:
|
4 |
+
- categorize extracted requirements,
|
5 |
+
- search and generate solutions solving those requirements,
|
6 |
+
- criticize those solutions
|
7 |
+
- refine the generated solutions based on the critics.
|
8 |
+
|
9 |
+
Please consult the OpenAPI doc for more information on the endpoints.
|