Lucas ARRIESSE commited on
Commit
c1e5d8a
·
1 Parent(s): 8cc6fe4
Files changed (2) hide show
  1. app.py +2 -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.