o
yf @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlZddlZddl m
Z
ddl m Z dZ
dd Zd
d Zd d
Zdd Zdd Zd ddZd ddZdd Zdd Zdd ZdS ) zGthis module contains code for processing response from fahamu client.py N)urljoin)quote)GeneNetworkQAClient)DocIDsz*https://genenetwork.fahamuai.com/api/tasksc C sN t | tr
| dS t | tr%| d d| d d| d d| d dS | S )z)Function for formatting bibliography infoz.txtauthor.titleyeardoi )
isinstancestr removesuffixdict)bib_info r >/home/shebes/code/GN-LLM-Hacking/gnqa_eval/src/apis/process.pyformat_bibliography_info s
,r c C s t ddd | D S )z1helper function for filtering non-printable chars c S s g | ] }|t jv rt|qS r )string printabler
).0charr r r
<listcomp> s z(filter_response_text.<locals>.<listcomp>)jsonloadsjoin)valr r r filter_response_text s
r c
C sf g }| D ]*\}}d}|D ]
}|d|d 7 }q||}||kr#|n||} ||| |d q|S )z!function to parse doc_ids contentr text)doc_idbibInfocomboTxt)itemsappend)
context
get_info_funcformat_bib_funcresultsdoc_idssummary combo_txtentrydoc_infor r r r
parse_context$ s
r/ c
C s z)t td| d| d| }dd| i}tj||d}| ddi| W S tjjyF } ztd |j j
d
|j j |d }~w t y[ } z
td t
| |d }~ww )
zJThis method is used to provide feedback for a document by making a rating.z/feedback?task_id=z
&document_id=z
&feedback=
AuthorizationzBearer )headersstatussuccessz HTTP Error Occurred: z -with status code- NzAn error occurred: )r BASE_URLrequestspostraise_for_statusr
exceptions HTTPError RuntimeErrorresponser status_code Exceptionr
) task_idr! rating
auth_tokenurlr1 resp
http_errorerrorr r r
rate_document3 s* rE c C sb t j|| }t j|st| dt|d
}t|W d S 1 s*w Y dS )z#function to open and load json filez was not found or is a directoryrbN)ospathr isfileFileNotFoundErroropenr load)filenamedir_path file_path file_handlerr r r load_fileF s $rQ r c C sj z"t |tj|d}| D ]}||d r ||d |d< q
| W S ty4 td|| | Y S w )z3method to fetch and populate references with pubmedz gn-meta/litr! pubmedz$failed to find pubmed_path for %s/%s)rQ rG rH r getrJ loggingrD )
references file_namedata_dirrR referencer r r fetch_pubmedO s rY c C s t t |d}|dt| |\}}|dkr!tdt| ||\}}|dkrVt|j }|
ddu r=|dg fS |d d }|d d
} t | t j
t}
|||
fS |dg fS ) z)entry function for the gn3 api endpoint()api_keyz?ask=r z Error connecting to Fahamu Api: dataN8Please try to rephrase your question to receive feedbackanswerr& )r r5 Sessionaskr r: r
get_answerr r rS r/ r getInfor ) queryr@ tmp_dir
api_clientresr> r3 resp_textr_ r& rU r r r get_gnqa_ s
ri c C s t t | d}||\}}|dkr=t|j}|dd u r$|dg fS |d d }|d d }t|t j t
}|||fS |dg fS )NrZ r\ r] r^ r_ r& ) r r5 r` r_ r r rS r/ r rc r ) r@ r> rf rg r3 rh r_ r& rU r r r get_response_from_taskidu s
rj c C s2 | d| d| }|rt|S | dg ddS )z,this method fetches prev user query searchesLLM:-zSorry No answer for youN)rd r_ rU r> )rS r r )rd user_id
redis_connresultr r r fetch_query_results s
rp c C s* | d| d}dd dd |D D S )z0methods to fetch all queries for a specific userrk *c S s g | ]}|d kr|qS r r )r rd r r r r s z$get_user_queries.<locals>.<listcomp>c S s g | ] }| d d qS )rl ) partition)r ro r r r r s )keys)rm rn r) r r r get_user_queries s rv rr )__doc__rG r r urllib.parser r rT r5 apis.gnqaclientr apis.respr r4 r r r/ rE rQ rY ri rj rp rv r r r r <module> s*
