3636from src .ragas_eval import *
3737from starlette .types import ASGIApp , Message , Receive , Scope , Send
3838import gzip
39- < << << << HEAD
4039from langchain_neo4j import Neo4jGraph
41- == == == =
42- >> >> >> > 6 a46472dcebfb9282c1c9d7332bcfacb30811661
4340
4441logger = CustomLogger ()
4542CHUNK_DIR = os .path .join (os .path .dirname (__file__ ), "chunks" )
@@ -85,16 +82,9 @@ async def __call__(self, scope: Scope, receive: Receive, send: Send):
8582 await gzip_middleware (scope , receive , send )
8683app = FastAPI ()
8784# SecWeb(app=app, Option={'referrer': False, 'xframe': False})
88- < << << << HEAD
8985# app.add_middleware(ContentSecurityPolicy, Option={'default-src': ["'self'"], 'base-uri': ["'self'"], 'block-all-mixed-content': []}, script_nonce=False, style_nonce=False, report_only=False)
9086app .add_middleware (XContentTypeOptions )
9187app .add_middleware (XFrame , Option = {'X-Frame-Options' : 'DENY' })
92- == == == =
93- app .add_middleware (ContentSecurityPolicy , Option = {'default-src' : ["'self'" ], 'base-uri' : ["'self'" ], 'block-all-mixed-content' : []}, script_nonce = False , style_nonce = False , report_only = False )
94- app .add_middleware (XContentTypeOptions )
95- app .add_middleware (XFrame , Option = {'X-Frame-Options' : 'DENY' })
96- #app.add_middleware(GZipMiddleware, minimum_size=1000, compresslevel=5)
97- > >> >> >> 6 a46472dcebfb9282c1c9d7332bcfacb30811661
9888app .add_middleware (CustomGZipMiddleware , minimum_size = 1000 , compresslevel = 5 ,paths = ["/sources_list" ,"/url/scan" ,"/extract" ,"/chat_bot" ,"/chunk_entities" ,"/get_neighbours" ,"/graph_query" ,"/schema" ,"/populate_graph_schema" ,"/get_unconnected_nodes_list" ,"/get_duplicate_nodes" ,"/fetch_chunktext" ])
9989app .add_middleware (
10090 CORSMiddleware ,
@@ -839,15 +829,11 @@ async def retry_processing(uri=Form(), userName=Form(), password=Form(), databas
839829 start = time .time ()
840830 graph = create_graph_database_connection (uri , userName , password , database )
841831 await asyncio .to_thread (set_status_retry , graph ,file_name ,retry_condition )
842- << << << < HEAD
843832 end = time .time ()
844833 elapsed_time = end - start
845834 json_obj = {'api_name' :'retry_processing' , 'db_url' :uri , 'userName' :userName , 'database' :database , 'file_name' :file_name ,'retry_condition' :retry_condition ,
846835 'logging_time' : formatted_time (datetime .now (timezone .utc )), 'elapsed_api_time' :f'{ elapsed_time :.2f} ' }
847836 logger .log_struct (json_obj , "INFO" )
848- == == == =
849- #set_status_retry(graph,file_name,retry_condition)
850- >> >> >> > 6 a46472dcebfb9282c1c9d7332bcfacb30811661
851837 return create_api_response ('Success' ,message = f"Status set to Ready to Reprocess for filename : { file_name } " )
852838 except Exception as e :
853839 job_status = "Failed"
@@ -963,11 +949,8 @@ async def fetch_chunktext(
963949 json_obj = {
964950 'api_name' : 'fetch_chunktext' ,
965951 'db_url' : uri ,
966- << << << < HEAD
967952 'userName' : userName ,
968953 'database' : database ,
969- == == == =
970- >> >> >> > 6 a46472dcebfb9282c1c9d7332bcfacb30811661
971954 'document_name' : document_name ,
972955 'page_no' : page_no ,
973956 'logging_time' : formatted_time (datetime .now (timezone .utc )),
@@ -985,7 +968,6 @@ async def fetch_chunktext(
985968 gc .collect ()
986969
987970
988- << << << < HEAD
989971@app .post ("/backend_connection_configuation" )
990972async def backend_connection_configuation ():
991973 try :
@@ -1015,7 +997,5 @@ async def backend_connection_configuation():
1015997 finally :
1016998 gc .collect ()
1017999
1018- == == == =
1019- >> >> >> > 6 a46472dcebfb9282c1c9d7332bcfacb30811661
10201000if __name__ == "__main__" :
10211001 uvicorn .run (app )
0 commit comments