rrg92 commited on
Commit
a986bad
·
1 Parent(s): 5d9faf7

Added endpoint specific

Browse files
Files changed (1) hide show
  1. app/main.py +17 -0
app/main.py CHANGED
@@ -42,3 +42,20 @@ async def InvokeGradioApi(request: Request, x_hf_token: str = Header(default=Non
42
 
43
 
44
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
42
 
43
 
44
 
45
+ @app.get("/SearchScripts")
46
+ async def SearchScript(request: Request, x_hf_token: str = Header(default=None)):
47
+ data = dict(request.query_params)
48
+
49
+ auth_key = data.pop("auth", None) or x_hf_token
50
+ client = Client('rrg92/sqlserver-lib-assistant', hf_token=auth_key if auth_key else None)
51
+
52
+ search_text = data['text'];
53
+
54
+ print("search", search_text);
55
+
56
+ try:
57
+ result = client.predict(message=search_text, api_name="/SearchFiles")
58
+ return result
59
+ except Exception as e:
60
+ print(e);
61
+ return {"error": "Some error happened. Admin must check logs"}