mgbam commited on
Commit
d7e83bd
·
verified ·
1 Parent(s): 2389422

Delete agent

Browse files
Files changed (2) hide show
  1. agent/clinical_nlp.py +0 -46
  2. agent/gemini_agent.py +0 -10
agent/clinical_nlp.py DELETED
@@ -1,46 +0,0 @@
1
- import requests
2
- import os
3
-
4
- def get_umls_api_key():
5
- return os.getenv("UMLS_API_KEY")
6
-
7
- def get_bioportal_api_key():
8
- return os.getenv("BIOPORTAL_API_KEY")
9
-
10
- def umls_concept_lookup(term):
11
- key = get_umls_api_key()
12
- if not key:
13
- return [{"error": "UMLS API key not set."}]
14
- url = f"https://uts-ws.nlm.nih.gov/rest/search/current?string={term}&apiKey={key}"
15
- try:
16
- response = requests.get(url, timeout=10)
17
- results = response.json().get("result", {}).get("results", [])
18
- return [
19
- {
20
- "name": r.get("name"),
21
- "ui": r.get("ui"),
22
- "rootSource": r.get("rootSource")
23
- }
24
- for r in results[:8]
25
- ]
26
- except Exception as e:
27
- return [{"error": f"UMLS error: {e}"}]
28
-
29
- def bioportal_concept_lookup(term):
30
- key = get_bioportal_api_key()
31
- if not key:
32
- return [{"error": "BioPortal API key not set."}]
33
- url = f"https://data.bioontology.org/search?q={term}&apikey={key}"
34
- try:
35
- response = requests.get(url, timeout=10)
36
- matches = response.json().get("collection", [])
37
- return [
38
- {
39
- "prefLabel": m.get("prefLabel"),
40
- "ontology": m.get("links", {}).get("ontology"),
41
- "id": m.get("@id")
42
- }
43
- for m in matches[:8]
44
- ]
45
- except Exception as e:
46
- return [{"error": f"BioPortal error: {e}"}]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
agent/gemini_agent.py DELETED
@@ -1,10 +0,0 @@
1
- import os, google.generativeai as genai
2
- from config.settings import settings
3
-
4
- genai.configure(api_key=settings.gemini_api_key)
5
-
6
- def chat_with_gemini(user, prompt):
7
- model = genai.GenerativeModel("gemini-pro")
8
- chat = model.start_chat()
9
- resp = chat.send_message(prompt)
10
- return resp.text