aiagents4pharma 1.40.1__py3-none-any.whl → 1.41.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- aiagents4pharma/talk2scholars/configs/agents/talk2scholars/s2_agent/default.yaml +4 -0
- aiagents4pharma/talk2scholars/configs/tools/question_and_answer/default.yaml +44 -4
- aiagents4pharma/talk2scholars/tests/test_nvidia_nim_reranker.py +127 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_answer_formatter.py +66 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_batch_processor.py +101 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_collection_manager.py +150 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_document_processor.py +69 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_generate_answer.py +75 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_gpu_detection.py +140 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_paper_loader.py +116 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_rag_pipeline.py +98 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_retrieve_chunks.py +197 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_singleton_manager.py +156 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_vector_normalization.py +121 -0
- aiagents4pharma/talk2scholars/tests/test_pdf_vector_store.py +434 -0
- aiagents4pharma/talk2scholars/tests/test_question_and_answer_tool.py +89 -509
- aiagents4pharma/talk2scholars/tests/test_tool_helper_utils.py +34 -89
- aiagents4pharma/talk2scholars/tools/paper_download/download_biorxiv_input.py +8 -6
- aiagents4pharma/talk2scholars/tools/paper_download/download_medrxiv_input.py +6 -4
- aiagents4pharma/talk2scholars/tools/pdf/question_and_answer.py +74 -40
- aiagents4pharma/talk2scholars/tools/pdf/utils/__init__.py +26 -1
- aiagents4pharma/talk2scholars/tools/pdf/utils/answer_formatter.py +62 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/batch_processor.py +200 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/collection_manager.py +172 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/document_processor.py +76 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/generate_answer.py +14 -14
- aiagents4pharma/talk2scholars/tools/pdf/utils/get_vectorstore.py +63 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/gpu_detection.py +154 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/nvidia_nim_reranker.py +60 -40
- aiagents4pharma/talk2scholars/tools/pdf/utils/paper_loader.py +123 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/rag_pipeline.py +122 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/retrieve_chunks.py +162 -40
- aiagents4pharma/talk2scholars/tools/pdf/utils/singleton_manager.py +140 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/tool_helper.py +40 -78
- aiagents4pharma/talk2scholars/tools/pdf/utils/vector_normalization.py +159 -0
- aiagents4pharma/talk2scholars/tools/pdf/utils/vector_store.py +277 -96
- aiagents4pharma/talk2scholars/tools/s2/multi_paper_rec.py +12 -9
- aiagents4pharma/talk2scholars/tools/s2/query_dataframe.py +0 -1
- aiagents4pharma/talk2scholars/tools/s2/retrieve_semantic_scholar_paper_id.py +9 -8
- aiagents4pharma/talk2scholars/tools/s2/single_paper_rec.py +5 -5
- {aiagents4pharma-1.40.1.dist-info → aiagents4pharma-1.41.0.dist-info}/METADATA +27 -115
- {aiagents4pharma-1.40.1.dist-info → aiagents4pharma-1.41.0.dist-info}/RECORD +45 -23
- aiagents4pharma/talk2scholars/tests/test_nvidia_nim_reranker_utils.py +0 -28
- {aiagents4pharma-1.40.1.dist-info → aiagents4pharma-1.41.0.dist-info}/WHEEL +0 -0
- {aiagents4pharma-1.40.1.dist-info → aiagents4pharma-1.41.0.dist-info}/licenses/LICENSE +0 -0
- {aiagents4pharma-1.40.1.dist-info → aiagents4pharma-1.41.0.dist-info}/top_level.txt +0 -0
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.4
|
2
2
|
Name: aiagents4pharma
|
3
|
-
Version: 1.
|
3
|
+
Version: 1.41.0
|
4
4
|
Summary: AI Agents for drug discovery, drug development, and other pharmaceutical R&D.
|
5
5
|
Classifier: Programming Language :: Python :: 3
|
6
6
|
Classifier: License :: OSI Approved :: MIT License
|
@@ -20,6 +20,7 @@ Requires-Dist: langchain==0.3.7
|
|
20
20
|
Requires-Dist: langchain-community==0.3.5
|
21
21
|
Requires-Dist: langchain-core==0.3.40
|
22
22
|
Requires-Dist: langchain-experimental==0.3.3
|
23
|
+
Requires-Dist: langchain-milvus==0.2.0
|
23
24
|
Requires-Dist: langchain-nvidia-ai-endpoints==0.3.9
|
24
25
|
Requires-Dist: langchain-openai==0.2.5
|
25
26
|
Requires-Dist: langchain_ollama==0.2.3
|
@@ -37,6 +38,7 @@ Requires-Dist: pypdf==5.2.0
|
|
37
38
|
Requires-Dist: pytest==8.3.3
|
38
39
|
Requires-Dist: pytest-asyncio==0.25.2
|
39
40
|
Requires-Dist: pyzotero==1.6.9
|
41
|
+
Requires-Dist: pymilvus==2.5.11
|
40
42
|
Requires-Dist: streamlit==1.39.0
|
41
43
|
Requires-Dist: sentence_transformers==3.3.1
|
42
44
|
Requires-Dist: tabulate==0.9.0
|
@@ -111,103 +113,12 @@ Our toolkit currently consists of the following agents:
|
|
111
113
|
|
112
114
|
_We now have all the agents available on Docker Hub._
|
113
115
|
|
114
|
-
|
115
|
-
|
116
|
-
If your machine has NVIDIA GPU(s), please install the following this:
|
117
|
-
- [nvidia-cuda-toolkit](https://developer.nvidia.com/cuda-toolkit)
|
118
|
-
- [nvidia-container-toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/1.17.8/install-guide.html) (required for GPU support with Docker; enables containers to access NVIDIA GPUs for accelerated computing). After installing `nvidia-container-toolkit`, please restart Docker to ensure GPU support is enabled.
|
119
|
-
|
120
|
-
##### **1. Download docker-compose.yml, .env.example and startup.sh from GitHub**
|
121
|
-
|
122
|
-
###### Talk2Agents4Pharma
|
123
|
-
|
124
|
-
```sh
|
125
|
-
mkdir talk2aiagents4pharma && cd talk2aiagents4pharma && wget https://raw.githubusercontent.com/VirtualPatientEngine/AIAgents4Pharma/main/aiagents4pharma/talk2aiagents4pharma/docker-compose.yml https://raw.githubusercontent.com/VirtualPatientEngine/AIAgents4Pharma/main/aiagents4pharma/talk2aiagents4pharma/.env.example https://raw.githubusercontent.com/VirtualPatientEngine/AIAgents4Pharma/main/aiagents4pharma/talk2aiagents4pharma/startup.sh
|
126
|
-
```
|
127
|
-
|
128
|
-
###### Talk2KnowledgeGraphs
|
129
|
-
|
130
|
-
```sh
|
131
|
-
mkdir talk2knowledgegraphs && cd talk2knowledgegraphs && wget https://raw.githubusercontent.com/VirtualPatientEngine/AIAgents4Pharma/main/aiagents4pharma/talk2knowledgegraphs/docker-compose.yml https://raw.githubusercontent.com/VirtualPatientEngine/AIAgents4Pharma/main/aiagents4pharma/talk2knowledgegraphs/.env.example https://raw.githubusercontent.com/VirtualPatientEngine/AIAgents4Pharma/main/aiagents4pharma/talk2knowledgegraphs/startup.sh
|
132
|
-
```
|
133
|
-
|
134
|
-
##### **2. Setup environment variables**
|
135
|
-
|
136
|
-
Copy and configure your `.env` file:
|
137
|
-
|
138
|
-
```sh
|
139
|
-
cp .env.example .env
|
140
|
-
```
|
141
|
-
|
142
|
-
Then edit `.env` and add your API keys:
|
143
|
-
|
144
|
-
```env
|
145
|
-
OPENAI_API_KEY=... # Required for both agents
|
146
|
-
NVIDIA_API_KEY=... # Required for both agents
|
147
|
-
LANGCHAIN_TRACING_V2=true # Optional for both agents
|
148
|
-
LANGCHAIN_API_KEY=... # Optional for both agents
|
149
|
-
```
|
150
|
-
|
151
|
-
[Additional Notes for Windows Users](https://github.com/VirtualPatientEngine/AIAgents4Pharma/blob/main/aiagents4pharma/talk2aiagents4pharma/install.md#notes-for-windows-users)
|
152
|
-
|
153
|
-
##### **3. Start the application**
|
154
|
-
|
155
|
-
```sh
|
156
|
-
chmod +x startup.sh
|
157
|
-
./startup.sh # Add --cpu flag to force CPU mode if needed
|
158
|
-
```
|
159
|
-
|
160
|
-
[More about startup script](https://github.com/VirtualPatientEngine/AIAgents4Pharma/blob/main/aiagents4pharma/talk2aiagents4pharma/install.md#about-startupsh)
|
161
|
-
|
162
|
-
##### **To Run Talk2Biomodels / Talk2Scholars**
|
163
|
-
|
164
|
-
###### Talk2Biomodels
|
165
|
-
|
166
|
-
```docker
|
167
|
-
docker run -d \
|
168
|
-
--name talk2biomodels \
|
169
|
-
-e OPENAI_API_KEY=<your_openai_api_key> \
|
170
|
-
-e NVIDIA_API_KEY=<your_nvidia_api_key> \
|
171
|
-
-p 8501:8501 \
|
172
|
-
virtualpatientengine/talk2biomodels
|
173
|
-
```
|
174
|
-
|
175
|
-
###### Talk2Scholars
|
176
|
-
|
177
|
-
```docker
|
178
|
-
docker run -d \
|
179
|
-
--name talk2scholars \
|
180
|
-
-e OPENAI_API_KEY=<your_openai_api_key> \
|
181
|
-
-e ZOTERO_API_KEY=<your_zotero_api_key> \
|
182
|
-
-e ZOTERO_USER_ID=<your_zotero_user_id> \
|
183
|
-
-e NVIDIA_API_KEY=<your_nvidia_api_key> \
|
184
|
-
-p 8501:8501 \
|
185
|
-
virtualpatientengine/talk2scholars
|
186
|
-
```
|
187
|
-
|
188
|
-
##### **4. Access the Web UI**
|
189
|
-
|
190
|
-
Once started, the agent is available at:
|
191
|
-
|
192
|
-
```
|
193
|
-
http://localhost:8501
|
194
|
-
```
|
195
|
-
|
196
|
-
To use **Talk2AIAgents4Pharma** or **Talk2KnowledgeGraphs**, you need a free **NVIDIA API key**. Create an account and apply for free credits [here](https://build.nvidia.com/explore/discover).
|
197
|
-
|
198
|
-
To use **Talk2BioModels** or **Talk2Scholars**, you need a free **NVIDIA API key**. Create an account and apply for free credits [here](https://build.nvidia.com/explore/discover).
|
199
|
-
|
200
|
-
Only for **Talk2Scholars**, you also need a **Zotero API key**, which you can generate [here](https://www.zotero.org/user/login#applications). _(For all other agents, the Zotero key is not required.)_
|
201
|
-
|
202
|
-
If you are using docker on Windows, please follow these [Windows Setup Notes](https://github.com/VirtualPatientEngine/AIAgents4Pharma/blob/main/aiagents4pharma/talk2aiagents4pharma/install.md#notes-for-windows-users).
|
203
|
-
|
204
|
-
**LangSmith** support is optional. To enable it, create an API key [here](https://docs.smith.langchain.com/administration/how_to_guides/organization_management/create_account_api_key).
|
205
|
-
|
206
|
-
[More on running multiple agents simultaneously](https://github.com/VirtualPatientEngine/AIAgents4Pharma/blob/main/aiagents4pharma/talk2aiagents4pharma/install.md#to-run-multiple-agents-simultaneously)
|
207
|
-
|
208
|
-
📝 By default, `talk2knowledgegraphs` includes a small subset of the PrimeKG knowledge graph, allowing users to start interacting with it out of the box.
|
209
|
-
To switch to a different knowledge graph or use your own, refer to the [deployment guide](https://virtualpatientengine.github.io/AIAgents4Pharma/talk2knowledgegraphs/deployment/).
|
116
|
+
Choose your agent below for detailed Docker instructions:
|
210
117
|
|
118
|
+
- [Talk2AIAgents4Pharma](aiagents4pharma/talk2aiagents4pharma/install.md)
|
119
|
+
- [Talk2KnowledgeGraphs](aiagents4pharma/talk2knowledgegraphs/install.md)
|
120
|
+
- [Talk2BioModels](aiagents4pharma/talk2biomodels/install.md)
|
121
|
+
- [Talk2Scholars](aiagents4pharma/talk2scholars/install.md)
|
211
122
|
|
212
123
|
#### Option 2: git (for developers and contributors)
|
213
124
|
|
@@ -238,23 +149,6 @@ export LANGCHAIN_TRACING_V2=true # Optional for all agents
|
|
238
149
|
export LANGCHAIN_API_KEY=... # Optional for all agents
|
239
150
|
```
|
240
151
|
|
241
|
-
To use **Talk2AIAgents4Pharma**, **Talk2BioModels**, **Talk2KnowledgeGraphs**, or **Talk2Scholars**, you need a free **NVIDIA API key**. Create an account and apply for free credits [here](https://build.nvidia.com/explore/discover).
|
242
|
-
|
243
|
-
Only for **Talk2Scholars**, you also need a **Zotero API key**, which you can generate [here](https://www.zotero.org/user/login#applications). _(For all other agents, the Zotero key is not required.)_
|
244
|
-
|
245
|
-
To use **Talk2Scholars**, you must have **FAISS** installed through **Conda**. Follow installation instructions for your OS [here](https://github.com/VirtualPatientEngine/AIAgents4Pharma/tree/main/aiagents4pharma/talk2scholars/install.md).
|
246
|
-
|
247
|
-
Additionally on **Windows**, the `pcst_fast 1.0.10` library requires **Microsoft Visual C++ 14.0 or greater**.
|
248
|
-
You can download the **Microsoft C++ Build Tools** [here](https://visualstudio.microsoft.com/visual-cpp-build-tools/).
|
249
|
-
|
250
|
-
**LangSmith** support is optional. To enable it, create an API key [here](https://docs.smith.langchain.com/administration/how_to_guides/organization_management/create_account_api_key).
|
251
|
-
|
252
|
-
_Please note that this will create a new tracing project in your Langsmith
|
253
|
-
account with the name `T2X-xxxx`, where `X` can be `AA4P` (Main Agent),
|
254
|
-
`B` (Biomodels), `S` (Scholars), `KG` (KnowledgeGraphs), or `C` (Cells).
|
255
|
-
If you skip the previous step, it will default to the name `default`.
|
256
|
-
`xxxx` will be the 4-digit ID created for the session._
|
257
|
-
|
258
152
|
4. **Launch the app:**
|
259
153
|
|
260
154
|
```sh
|
@@ -269,10 +163,28 @@ _Replace `<agent>` with the agent name you are interested to launch:_
|
|
269
163
|
- `talk2scholars`
|
270
164
|
- `talk2cells`
|
271
165
|
|
166
|
+
If your machine has NVIDIA GPU(s), please install the following this:
|
167
|
+
|
168
|
+
- [nvidia-cuda-toolkit](https://developer.nvidia.com/cuda-toolkit)
|
169
|
+
- [nvidia-container-toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/1.17.8/install-guide.html) (required for GPU support with Docker; enables containers to access NVIDIA GPUs for accelerated computing). After installing `nvidia-container-toolkit`, please restart Docker to ensure GPU support is enabled.
|
170
|
+
|
171
|
+
To use the **Agents**, you need a free **NVIDIA API key**. Create an account and apply for free credits [here](https://build.nvidia.com/explore/discover).
|
172
|
+
|
173
|
+
**Talk2Scholars** requires Milvus to be set up as the vector database — install Milvus depending on your setup by following the official instructions for [CPU](https://milvus.io/docs/install_standalone-docker-compose.md) or [GPU](https://milvus.io/docs/install_standalone-docker-compose-gpu.md). You will also need a **Zotero API key**, which you can generate [here](https://www.zotero.org/user/login#applications). _(The Zotero key is only required for Talk2Scholars; all other agents do not need it.)_
|
174
|
+
|
175
|
+
> Additionally on **Windows**, the `pcst_fast 1.0.10` library requires **Microsoft Visual C++ 14.0 or greater**.
|
176
|
+
> You can download the **Microsoft C++ Build Tools** [here](https://visualstudio.microsoft.com/visual-cpp-build-tools/).
|
177
|
+
|
272
178
|
📝 By default, `talk2knowledgegraphs` includes a small subset of the PrimeKG knowledge graph, allowing users to start interacting with it out of the box.
|
273
179
|
To switch to a different knowledge graph or use your own, refer to the [deployment guide](https://virtualpatientengine.github.io/AIAgents4Pharma/talk2knowledgegraphs/deployment/).
|
274
180
|
|
275
|
-
|
181
|
+
**LangSmith** support is optional. To enable it, create an API key [here](https://docs.smith.langchain.com/administration/how_to_guides/organization_management/create_account_api_key).
|
182
|
+
|
183
|
+
_Please note that this will create a new tracing project in your Langsmith
|
184
|
+
account with the name `T2X-xxxx`, where `X` can be `AA4P` (Main Agent),
|
185
|
+
`B` (Biomodels), `S` (Scholars), `KG` (KnowledgeGraphs), or `C` (Cells).
|
186
|
+
If you skip the previous step, it will default to the name `default`.
|
187
|
+
`xxxx` will be the 4-digit ID created for the session._
|
276
188
|
|
277
189
|
#### Option 3: pip (beta-release)
|
278
190
|
|
@@ -167,7 +167,7 @@ aiagents4pharma/talk2scholars/configs/agents/talk2scholars/paper_download_agent/
|
|
167
167
|
aiagents4pharma/talk2scholars/configs/agents/talk2scholars/pdf_agent/__init__.py,sha256=fqQQ-GlRcbzru2KmEk3oMma0R6_SzGM8dOXzYeU4oVA,46
|
168
168
|
aiagents4pharma/talk2scholars/configs/agents/talk2scholars/pdf_agent/default.yaml,sha256=p3hFJi-1xbHCNagTG13WHvsrPt11XQg1nHS7VJx2XOY,166
|
169
169
|
aiagents4pharma/talk2scholars/configs/agents/talk2scholars/s2_agent/__init__.py,sha256=fqQQ-GlRcbzru2KmEk3oMma0R6_SzGM8dOXzYeU4oVA,46
|
170
|
-
aiagents4pharma/talk2scholars/configs/agents/talk2scholars/s2_agent/default.yaml,sha256=
|
170
|
+
aiagents4pharma/talk2scholars/configs/agents/talk2scholars/s2_agent/default.yaml,sha256=CHMsW3cpGtfYWpSIY5XgNEYPvbI31OjOYfl_VbbKIyY,707
|
171
171
|
aiagents4pharma/talk2scholars/configs/agents/talk2scholars/zotero_agent/__init__.py,sha256=fqQQ-GlRcbzru2KmEk3oMma0R6_SzGM8dOXzYeU4oVA,46
|
172
172
|
aiagents4pharma/talk2scholars/configs/agents/talk2scholars/zotero_agent/default.yaml,sha256=FUbJIiH_FHeL0n1M-HSsLX_R2gMBRxBnPcejhd0HrYc,399
|
173
173
|
aiagents4pharma/talk2scholars/configs/app/__init__.py,sha256=tXpOW3R4eAfNoqvoaHfabSG-DcMHmUGSTg_4zH_vlgw,94
|
@@ -183,7 +183,7 @@ aiagents4pharma/talk2scholars/configs/tools/download_medrxiv_paper/default.yaml,
|
|
183
183
|
aiagents4pharma/talk2scholars/configs/tools/multi_paper_recommendation/__init__.py,sha256=fqQQ-GlRcbzru2KmEk3oMma0R6_SzGM8dOXzYeU4oVA,46
|
184
184
|
aiagents4pharma/talk2scholars/configs/tools/multi_paper_recommendation/default.yaml,sha256=comNgL9hRpH--IWuEsrN6hV5WdrJmh-ZsRh7hbryVhg,631
|
185
185
|
aiagents4pharma/talk2scholars/configs/tools/question_and_answer/__init__.py,sha256=fqQQ-GlRcbzru2KmEk3oMma0R6_SzGM8dOXzYeU4oVA,46
|
186
|
-
aiagents4pharma/talk2scholars/configs/tools/question_and_answer/default.yaml,sha256=
|
186
|
+
aiagents4pharma/talk2scholars/configs/tools/question_and_answer/default.yaml,sha256=Pa3JVyA9tabXZ4Bk3n5dAml7P-nXUcT7HgkA8Kr_sXk,2238
|
187
187
|
aiagents4pharma/talk2scholars/configs/tools/retrieve_semantic_scholar_paper_id/__init__.py,sha256=fqQQ-GlRcbzru2KmEk3oMma0R6_SzGM8dOXzYeU4oVA,46
|
188
188
|
aiagents4pharma/talk2scholars/configs/tools/retrieve_semantic_scholar_paper_id/default.yaml,sha256=HG-N8yRjlX9zFwbIBvaDI9ndKjfL-gqPTCCPMLgdUpw,271
|
189
189
|
aiagents4pharma/talk2scholars/configs/tools/search/__init__.py,sha256=fqQQ-GlRcbzru2KmEk3oMma0R6_SzGM8dOXzYeU4oVA,46
|
@@ -198,13 +198,25 @@ aiagents4pharma/talk2scholars/state/__init__.py,sha256=ReScKLpEvedq4P6ww52NRQS0X
|
|
198
198
|
aiagents4pharma/talk2scholars/state/state_talk2scholars.py,sha256=Z2zV-SXB2SMnn8PnjWjmK-OD5KjUwMTChBpXBAcl2hg,3885
|
199
199
|
aiagents4pharma/talk2scholars/tests/__init__.py,sha256=U3PsTiUZaUBD1IZanFGkDIOdFieDVJtGKQ5-woYUo8c,45
|
200
200
|
aiagents4pharma/talk2scholars/tests/test_main_agent.py,sha256=4Z3xLq8MGlayGhQE5qKOirYotwJrlf7fk8rqAaORorg,7617
|
201
|
-
aiagents4pharma/talk2scholars/tests/
|
201
|
+
aiagents4pharma/talk2scholars/tests/test_nvidia_nim_reranker.py,sha256=ftfn4VenufZ-zt2nSOyOoCjfSbwFOW9CS1DLzPtBEaM,4410
|
202
202
|
aiagents4pharma/talk2scholars/tests/test_paper_download_agent.py,sha256=gKSQp-sw62FplNnGYW0wv2ZIUEefh3o0tFWbRzy9yLs,5068
|
203
203
|
aiagents4pharma/talk2scholars/tests/test_paper_download_biorxiv.py,sha256=gosuW4VBXyorQXbf0TpgAIT2hQjEeuvTTnT1jnoBYqM,6405
|
204
204
|
aiagents4pharma/talk2scholars/tests/test_paper_download_medrxiv.py,sha256=iNq9vEIVapmnUZTRJXCv_UoaWThGapW7Vt_2BmZG9NE,6414
|
205
205
|
aiagents4pharma/talk2scholars/tests/test_paper_download_tools.py,sha256=lGXbHl3lEXDjMHAX9uCgrREBOUuOHWv9TsYEshiG_tc,10421
|
206
206
|
aiagents4pharma/talk2scholars/tests/test_pdf_agent.py,sha256=9Kr0FcyFWmUDTasYh6ZdS-OWQqy37mH9K3p5Y0dqQHw,4283
|
207
|
-
aiagents4pharma/talk2scholars/tests/
|
207
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_answer_formatter.py,sha256=a1a_z1M9sOQ_SFo-gHM3xA_f5MoJJmEoW8Tc3AX9vL0,2239
|
208
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_batch_processor.py,sha256=e8KQJ80nbOmCH4mgDnIXXjlrRk0zf0g-5Odt2jsCU0s,3251
|
209
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_collection_manager.py,sha256=M7mT_t2X4DUw-oSicnUnRU73exVDldQIcHlwQmIyV2s,5147
|
210
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_document_processor.py,sha256=G_LPOqfIDAah-24GScIaXu6-McpddLd-1i779bbJg-c,2576
|
211
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_generate_answer.py,sha256=h2MITyWsaRWrW3vxTGKht0JyhEfRW0Tomhyq257eNrY,2657
|
212
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_gpu_detection.py,sha256=nh8VyzqMEIQ3IjbTYSX29tjqLLCXmmHY74AQG6ZOUis,4952
|
213
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_paper_loader.py,sha256=QOwFnKN-fl6h7b_yaOfos96G0qR4AxQvyJfH6TK4u_Q,3678
|
214
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_rag_pipeline.py,sha256=DfiyzVlx-ZXQK99C2ieVgVafL7ujcC9k2dRPZ82tTNk,2949
|
215
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_retrieve_chunks.py,sha256=2cjbCrf86z6kajV-8QDEW_J0ZWtzA3jRILaIscfsGyI,7729
|
216
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_singleton_manager.py,sha256=gOk8L9wNRPUnAiB89n1a4pQQPI7UtonRi7-IcCip94k,5628
|
217
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_vector_normalization.py,sha256=pu9I3tromjToIN5r4S8sWanaLBVhhk71UPP5zTTGwZY,3928
|
218
|
+
aiagents4pharma/talk2scholars/tests/test_pdf_vector_store.py,sha256=-7CUiPLT4mOBVkNV2qlF7t4yU_mRArM0uAcUAKgEY4k,15644
|
219
|
+
aiagents4pharma/talk2scholars/tests/test_question_and_answer_tool.py,sha256=UKSD7wLJ1cZunw8a1BHM7E-NUML6blxpqoDroCYuvQs,4245
|
208
220
|
aiagents4pharma/talk2scholars/tests/test_read_helper_utils.py,sha256=yTT1aLpTydDSdGcRZur5cMktwYZbFK5NEUgOBvltcWg,3819
|
209
221
|
aiagents4pharma/talk2scholars/tests/test_s2_agent.py,sha256=TsdNlZ6vHz18bbX6Vto28nbBLRDI94wSFt5-1acDK64,7768
|
210
222
|
aiagents4pharma/talk2scholars/tests/test_s2_display_dataframe.py,sha256=2VPPZitQRWDZV0ceaK2-hQqkIvoigSPWNHoFOgKJjQE,3107
|
@@ -215,7 +227,7 @@ aiagents4pharma/talk2scholars/tests/test_s2_search.py,sha256=mCGpoCYVn0SJ9BPcEjT
|
|
215
227
|
aiagents4pharma/talk2scholars/tests/test_s2_single.py,sha256=KjSh7V2cl1IuO_M9O6dj0vnMHr13H-xKxia_ZgT4qag,10313
|
216
228
|
aiagents4pharma/talk2scholars/tests/test_s2_utils_ext_ids.py,sha256=6xbHzClkYI_ZcR3-Xl7nGs8-hB0IyLlxn8rzeXrJxFQ,15129
|
217
229
|
aiagents4pharma/talk2scholars/tests/test_state.py,sha256=A2lqA4h37QptLnwKWwm1Y79yELE4wtEBXzCiQ13YdLw,1270
|
218
|
-
aiagents4pharma/talk2scholars/tests/test_tool_helper_utils.py,sha256=
|
230
|
+
aiagents4pharma/talk2scholars/tests/test_tool_helper_utils.py,sha256=64Z2JggDk-S8I00CNR69os8NGVLu63zd5pEJmMhC2wc,3365
|
219
231
|
aiagents4pharma/talk2scholars/tests/test_zotero_agent.py,sha256=iWbWlat5RWE8mmCSqKGNG7Xzbdieua6cKGq-jwqP4ws,6119
|
220
232
|
aiagents4pharma/talk2scholars/tests/test_zotero_human_in_the_loop.py,sha256=YelLQu9Y_r1SNQsC1xoLHJoJ3soIZtBt1MFbbNhY-Dg,10744
|
221
233
|
aiagents4pharma/talk2scholars/tests/test_zotero_path.py,sha256=Ko0HyXCrpm-vs8Bkf-syxp3MfL1IvZwXXgPExyQy_F8,18618
|
@@ -225,23 +237,33 @@ aiagents4pharma/talk2scholars/tests/test_zotero_write.py,sha256=qWlO0XoZJ6vxUxgi
|
|
225
237
|
aiagents4pharma/talk2scholars/tools/__init__.py,sha256=c8pYHDqR9P0Frz2jWjbvyizfSTBMlMFzGsiQzx2KC9c,189
|
226
238
|
aiagents4pharma/talk2scholars/tools/paper_download/__init__.py,sha256=Lu5FmBxDH8mIIYE41G8_BKYXUf-vHIYVwujidbeydl4,295
|
227
239
|
aiagents4pharma/talk2scholars/tools/paper_download/download_arxiv_input.py,sha256=e3S8JusJVVSEC_tFh8H5CS9ZqD20jX8hkFO9EYESBXo,5653
|
228
|
-
aiagents4pharma/talk2scholars/tools/paper_download/download_biorxiv_input.py,sha256=
|
229
|
-
aiagents4pharma/talk2scholars/tools/paper_download/download_medrxiv_input.py,sha256=
|
240
|
+
aiagents4pharma/talk2scholars/tools/paper_download/download_biorxiv_input.py,sha256=Iipvv0P-KLbW8ELMDBLr-FxgoGBjRjPToVw7pBVJS-Y,3676
|
241
|
+
aiagents4pharma/talk2scholars/tools/paper_download/download_medrxiv_input.py,sha256=2ONhx7rygCACf_SYc1cxDAmQP4ZzkB39OoLZYsbFMyI,3752
|
230
242
|
aiagents4pharma/talk2scholars/tools/pdf/__init__.py,sha256=DPpOfON3AySko5EBBAe_3udOoSaAdQWNyGeNvJyV5R8,138
|
231
|
-
aiagents4pharma/talk2scholars/tools/pdf/question_and_answer.py,sha256=
|
232
|
-
aiagents4pharma/talk2scholars/tools/pdf/utils/__init__.py,sha256=
|
233
|
-
aiagents4pharma/talk2scholars/tools/pdf/utils/
|
234
|
-
aiagents4pharma/talk2scholars/tools/pdf/utils/
|
235
|
-
aiagents4pharma/talk2scholars/tools/pdf/utils/
|
236
|
-
aiagents4pharma/talk2scholars/tools/pdf/utils/
|
237
|
-
aiagents4pharma/talk2scholars/tools/pdf/utils/
|
243
|
+
aiagents4pharma/talk2scholars/tools/pdf/question_and_answer.py,sha256=_HLO04bzTSlnEEWwkJ3rC9Gjz8MPeHVglqovoTDOSp4,5844
|
244
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/__init__.py,sha256=AlvoJY0hI4MCS4zHO3EtFkFJpKqjO7ZxWkoE3QvHi88,820
|
245
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/answer_formatter.py,sha256=E8XCA-R744nSgkSl7OIBdcAHjXLMFGaeZAFzOY4TPYI,1645
|
246
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/batch_processor.py,sha256=4lCQ4TOOIca3zZg4mF4uCQuRE5A2dI0Z7Kk1Z5y3Y2I,5893
|
247
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/collection_manager.py,sha256=CSYim0zZ_S0MNWecTz5U8KEX1L9Rtu1Ki8Kp_aX73iQ,5698
|
248
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/document_processor.py,sha256=-9o9OE6O0WpHw55Ps2RhR-uDjJlMDbYTnU2ZueS9KAw,2586
|
249
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/generate_answer.py,sha256=zTXo09zfGhPeU1IdfF8sk6lFOkx0GKOQrhbEnKGQujw,3299
|
250
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/get_vectorstore.py,sha256=u90DHqckgpC4BftzKRyd998GM5VG294ybA6Xby5clXs,2069
|
251
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/gpu_detection.py,sha256=VkTBkAWrV2RekO0swnxrRmbG5YtP3-Rl3fu2CaZ1b_Y,5402
|
252
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/nvidia_nim_reranker.py,sha256=k800UKpG7krqsfkwX_cHBm0Einu7MbmIwhi8Smj_nsM,3407
|
253
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/paper_loader.py,sha256=gMZasRX6XT-EbZ9EQlhB0un6bcFNbQb_xkQ76rhxQfQ,4060
|
254
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/rag_pipeline.py,sha256=avsw40v5MwoWrXKbiT0krLmwCY_5mYtCc-_LDYf-0ZY,3492
|
255
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/retrieve_chunks.py,sha256=u16hklQnGG7Mp0uBzpFDwV3PwAn6FtCrkabjhenw2Lg,6917
|
256
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/singleton_manager.py,sha256=KIz0ypThzaHyUHCbcg_mchW9Qr1uFK4OsmhD2NLUYZo,5052
|
257
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/tool_helper.py,sha256=NCc778mfWD4OAeA6On2me1c6n9YfFWujsd5h2L2L568,3034
|
258
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/vector_normalization.py,sha256=f4erq3JJW7Fdk7vM1qQf3i1si6FZzPoFHVfV98U3b7Y,4675
|
259
|
+
aiagents4pharma/talk2scholars/tools/pdf/utils/vector_store.py,sha256=X9OGwV3RHZIjglIbYJ7sK5feUAOfITgXdFRCcVjelXg,12817
|
238
260
|
aiagents4pharma/talk2scholars/tools/s2/__init__.py,sha256=w_eiw0pG8HNp79F9O_icXs_Yl_4odsmagYNKDTjIsvk,428
|
239
261
|
aiagents4pharma/talk2scholars/tools/s2/display_dataframe.py,sha256=qnY7AQDnAs0SrmV7AZ9pWm10HEmPlO7EBfzYvpb3jvs,3965
|
240
|
-
aiagents4pharma/talk2scholars/tools/s2/multi_paper_rec.py,sha256=
|
241
|
-
aiagents4pharma/talk2scholars/tools/s2/query_dataframe.py,sha256=
|
242
|
-
aiagents4pharma/talk2scholars/tools/s2/retrieve_semantic_scholar_paper_id.py,sha256=
|
262
|
+
aiagents4pharma/talk2scholars/tools/s2/multi_paper_rec.py,sha256=TEt4jOX1u8v3w_u10sGx-Ghwhe4yjuaYmUjD62nJQJM,3886
|
263
|
+
aiagents4pharma/talk2scholars/tools/s2/query_dataframe.py,sha256=EJ-EV-pk2kLS12Q5HatBdGue3t-MB8mx5PrSxtLN7Co,7042
|
264
|
+
aiagents4pharma/talk2scholars/tools/s2/retrieve_semantic_scholar_paper_id.py,sha256=0yILvpqraXldd61nLHSCfITvKyLIxts6HBC5GaGVfBQ,3995
|
243
265
|
aiagents4pharma/talk2scholars/tools/s2/search.py,sha256=SUAN32x1d9dNikFKitcXZZ0BhFfsGMdLDk0z0DpJXuA,3334
|
244
|
-
aiagents4pharma/talk2scholars/tools/s2/single_paper_rec.py,sha256=
|
266
|
+
aiagents4pharma/talk2scholars/tools/s2/single_paper_rec.py,sha256=JTggOB4sqhTF1kP81Gl_54RjpbHIfqtJEoUTxlZ82N8,3630
|
245
267
|
aiagents4pharma/talk2scholars/tools/s2/utils/__init__.py,sha256=wBTPVgiXbmIJUMouOQRwojgk5PJXeEinDJzHzEToZbU,229
|
246
268
|
aiagents4pharma/talk2scholars/tools/s2/utils/multi_helper.py,sha256=Z5G4e7R7cUO60_HYbcd3BJC_-jtybc5DGcOC8yjpprY,8642
|
247
269
|
aiagents4pharma/talk2scholars/tools/s2/utils/search_helper.py,sha256=wVkQW2KTmS2av4W5PqqRhCb53n-egZLpAD95xV2mO68,7839
|
@@ -256,8 +278,8 @@ aiagents4pharma/talk2scholars/tools/zotero/utils/review_helper.py,sha256=IPD1V9y
|
|
256
278
|
aiagents4pharma/talk2scholars/tools/zotero/utils/write_helper.py,sha256=ALwLecy1QVebbsmXJiDj1GhGmyhq2R2tZlAyEl1vfhw,7410
|
257
279
|
aiagents4pharma/talk2scholars/tools/zotero/utils/zotero_path.py,sha256=oIrfbOySgts50ksHKyjcWjRkPRIS88g3Lc0v9mBkU8w,6375
|
258
280
|
aiagents4pharma/talk2scholars/tools/zotero/utils/zotero_pdf_downloader.py,sha256=ERBha8afU6Q1EaRBe9qB8tchOzZ4_KfFgDW6EElOJoU,4816
|
259
|
-
aiagents4pharma-1.
|
260
|
-
aiagents4pharma-1.
|
261
|
-
aiagents4pharma-1.
|
262
|
-
aiagents4pharma-1.
|
263
|
-
aiagents4pharma-1.
|
281
|
+
aiagents4pharma-1.41.0.dist-info/licenses/LICENSE,sha256=IcIbyB1Hyk5ZDah03VNQvJkbNk2hkBCDqQ8qtnCvB4Q,1077
|
282
|
+
aiagents4pharma-1.41.0.dist-info/METADATA,sha256=vZTucUK52stvfzDaekMcwqo7SJC4QKhsta97q7iHQlY,12064
|
283
|
+
aiagents4pharma-1.41.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
284
|
+
aiagents4pharma-1.41.0.dist-info/top_level.txt,sha256=-AH8rMmrSnJtq7HaAObS78UU-cTCwvX660dSxeM7a0A,16
|
285
|
+
aiagents4pharma-1.41.0.dist-info/RECORD,,
|
@@ -1,28 +0,0 @@
|
|
1
|
-
"""
|
2
|
-
Unit tests for NVIDIA NIM reranker error handling in nvidia_nim_reranker.py
|
3
|
-
"""
|
4
|
-
|
5
|
-
import unittest
|
6
|
-
from types import SimpleNamespace
|
7
|
-
|
8
|
-
from aiagents4pharma.talk2scholars.tools.pdf.utils.nvidia_nim_reranker import (
|
9
|
-
rank_papers_by_query,
|
10
|
-
)
|
11
|
-
|
12
|
-
|
13
|
-
class TestNVIDIARerankerError(unittest.TestCase):
|
14
|
-
"""Tests for NVIDIA NIM reranker error handling."""
|
15
|
-
|
16
|
-
def test_missing_api_key_raises_value_error(self):
|
17
|
-
"""Ensure missing API key triggers ValueError."""
|
18
|
-
vector_store = SimpleNamespace(documents={})
|
19
|
-
# Config without API key
|
20
|
-
cfg = SimpleNamespace(
|
21
|
-
reranker=SimpleNamespace(model="m", api_key=None), top_k_papers=3
|
22
|
-
)
|
23
|
-
with self.assertRaises(ValueError) as cm:
|
24
|
-
rank_papers_by_query(vector_store, "query", cfg, top_k=cfg.top_k_papers)
|
25
|
-
self.assertEqual(
|
26
|
-
str(cm.exception),
|
27
|
-
"Configuration 'reranker.api_key' must be set for reranking",
|
28
|
-
)
|
File without changes
|
File without changes
|
File without changes
|