aiagents4pharma 1.47.0__py3-none-any.whl → 1.48.1__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/talk2aiagents4pharma/Dockerfile +29 -1
- aiagents4pharma/talk2aiagents4pharma/docker-compose/cpu/.env.example +2 -2
- aiagents4pharma/talk2aiagents4pharma/docker-compose/cpu/docker-compose.yml +1 -1
- aiagents4pharma/talk2aiagents4pharma/docker-compose/gpu/.env.example +2 -2
- aiagents4pharma/talk2aiagents4pharma/docker-compose/gpu/docker-compose.yml +1 -1
- aiagents4pharma/talk2aiagents4pharma/install.md +32 -1
- aiagents4pharma/talk2knowledgegraphs/Dockerfile +32 -4
- aiagents4pharma/talk2knowledgegraphs/docker-compose/cpu/.env.example +2 -2
- aiagents4pharma/talk2knowledgegraphs/docker-compose/cpu/docker-compose.yml +1 -1
- aiagents4pharma/talk2knowledgegraphs/docker-compose/gpu/.env.example +2 -2
- aiagents4pharma/talk2knowledgegraphs/docker-compose/gpu/docker-compose.yml +1 -1
- aiagents4pharma/talk2knowledgegraphs/install.md +32 -1
- aiagents4pharma/talk2knowledgegraphs/milvus_data_dump.py +30 -20
- {aiagents4pharma-1.47.0.dist-info → aiagents4pharma-1.48.1.dist-info}/METADATA +1 -1
- {aiagents4pharma-1.47.0.dist-info → aiagents4pharma-1.48.1.dist-info}/RECORD +17 -17
- {aiagents4pharma-1.47.0.dist-info → aiagents4pharma-1.48.1.dist-info}/WHEEL +0 -0
- {aiagents4pharma-1.47.0.dist-info → aiagents4pharma-1.48.1.dist-info}/licenses/LICENSE +0 -0
@@ -33,6 +33,7 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
|
|
33
33
|
&& update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1
|
34
34
|
|
35
35
|
FROM python-install AS uv-install
|
36
|
+
ARG INSTALL_CUDA=true
|
36
37
|
WORKDIR /app
|
37
38
|
|
38
39
|
# Install UV package manager and dependencies
|
@@ -42,14 +43,19 @@ RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
|
|
42
43
|
export UV_PROJECT_ENVIRONMENT="/opt/venv" && \
|
43
44
|
uv sync --frozen --extra dev --no-install-project --python python${PYTHON_VERSION} && \
|
44
45
|
. /opt/venv/bin/activate && \
|
46
|
+
if [ "$INSTALL_CUDA" = "true" ]; then \
|
45
47
|
uv pip install \
|
46
48
|
--extra-index-url=https://pypi.nvidia.com \
|
47
49
|
--index-strategy unsafe-best-match \
|
48
|
-
cudf-cu12 dask-cudf-cu12
|
50
|
+
cudf-cu12 dask-cudf-cu12; \
|
51
|
+
else \
|
52
|
+
echo "Skipping RAPIDS packages for CPU build"; \
|
53
|
+
fi && \
|
49
54
|
uv cache clean
|
50
55
|
|
51
56
|
FROM ${BASE_IMAGE} AS runtime
|
52
57
|
ARG PYTHON_VERSION=3.12
|
58
|
+
ARG INSTALL_CUDA=true
|
53
59
|
LABEL maintainer="talk2aiagents4pharma"
|
54
60
|
LABEL version="1.0.0"
|
55
61
|
LABEL description="AI Agents for Pharma - Streamlit Application"
|
@@ -58,6 +64,7 @@ LABEL description="AI Agents for Pharma - Streamlit Application"
|
|
58
64
|
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
59
65
|
ca-certificates \
|
60
66
|
curl \
|
67
|
+
gnupg \
|
61
68
|
libmagic1 \
|
62
69
|
libopenblas0 \
|
63
70
|
libomp5 \
|
@@ -66,6 +73,27 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
|
|
66
73
|
&& update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
|
67
74
|
&& update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1
|
68
75
|
|
76
|
+
# Install CUDA runtime libraries required by cudf/cupy (optional)
|
77
|
+
RUN if [ "$INSTALL_CUDA" = "true" ]; then \
|
78
|
+
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/3bf863cc.pub \
|
79
|
+
| gpg --dearmor -o /usr/share/keyrings/nvidia-cuda-keyring.gpg && \
|
80
|
+
echo "deb [signed-by=/usr/share/keyrings/nvidia-cuda-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/ /" \
|
81
|
+
> /etc/apt/sources.list.d/nvidia-cuda.list && \
|
82
|
+
apt-get update && \
|
83
|
+
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
84
|
+
cuda-cudart-12-6 \
|
85
|
+
cuda-cudart-dev-12-6 \
|
86
|
+
cuda-nvrtc-12-6 \
|
87
|
+
cuda-nvrtc-dev-12-6 \
|
88
|
+
libcublas-12-6 \
|
89
|
+
libcusparse-12-6 \
|
90
|
+
&& rm -rf /var/lib/apt/lists/*; \
|
91
|
+
else \
|
92
|
+
echo "Skipping CUDA installation"; \
|
93
|
+
fi
|
94
|
+
|
95
|
+
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64
|
96
|
+
|
69
97
|
# Copy UV virtual environment from build stage
|
70
98
|
COPY --from=uv-install /opt/venv /opt/venv
|
71
99
|
|
@@ -11,11 +11,11 @@ LANGCHAIN_API_KEY=your_langchain_api_key_here
|
|
11
11
|
NVIDIA_API_KEY=your_nvidia_api_key_here
|
12
12
|
|
13
13
|
# Set environment variables for data loader
|
14
|
-
MILVUS_HOST=
|
14
|
+
MILVUS_HOST=milvus-standalone
|
15
15
|
MILVUS_PORT=19530
|
16
16
|
MILVUS_USER=root
|
17
17
|
MILVUS_PASSWORD=Milvus
|
18
|
-
MILVUS_DATABASE=
|
18
|
+
MILVUS_DATABASE=t2kg_primekg
|
19
19
|
|
20
20
|
# Specify the data directory for multimodal data to your own data directory
|
21
21
|
# DATA_DIR=/your_absolute_path_to_your_data_dir/
|
@@ -4,7 +4,7 @@ services:
|
|
4
4
|
# talk2aiagents4pharma with automatic data loading via entrypoint
|
5
5
|
talk2aiagents4pharma:
|
6
6
|
container_name: talk2aiagents4pharma
|
7
|
-
image: vpatientengine/talk2aiagents4pharma:latest
|
7
|
+
image: vpatientengine/talk2aiagents4pharma:latest-cpu
|
8
8
|
platform: linux/amd64
|
9
9
|
ports:
|
10
10
|
- "8501:8501"
|
@@ -11,11 +11,11 @@ LANGCHAIN_API_KEY=your_langchain_api_key_here
|
|
11
11
|
NVIDIA_API_KEY=your_nvidia_api_key_here
|
12
12
|
|
13
13
|
# Set environment variables for data loader
|
14
|
-
MILVUS_HOST=
|
14
|
+
MILVUS_HOST=milvus-standalone
|
15
15
|
MILVUS_PORT=19530
|
16
16
|
MILVUS_USER=root
|
17
17
|
MILVUS_PASSWORD=Milvus
|
18
|
-
MILVUS_DATABASE=
|
18
|
+
MILVUS_DATABASE=t2kg_primekg
|
19
19
|
|
20
20
|
# Specify the data directory for multimodal data to your own data directory
|
21
21
|
# DATA_DIR=/your_absolute_path_to_your_data_dir/
|
@@ -4,7 +4,7 @@ services:
|
|
4
4
|
# talk2aiagents4pharma with automatic data loading via entrypoint
|
5
5
|
talk2aiagents4pharma:
|
6
6
|
container_name: talk2aiagents4pharma
|
7
|
-
image: vpatientengine/talk2aiagents4pharma:latest
|
7
|
+
image: vpatientengine/talk2aiagents4pharma:latest-gpu
|
8
8
|
platform: linux/amd64
|
9
9
|
ports:
|
10
10
|
- "8501:8501"
|
@@ -53,7 +53,7 @@ LANGCHAIN_API_KEY=your_langchain_api_key_here
|
|
53
53
|
NVIDIA_API_KEY=your_nvidia_api_key_here
|
54
54
|
|
55
55
|
# Set environment variables for data loader
|
56
|
-
MILVUS_HOST=
|
56
|
+
MILVUS_HOST=milvus-standalone
|
57
57
|
MILVUS_PORT=19530
|
58
58
|
MILVUS_USER=root
|
59
59
|
MILVUS_PASSWORD=Milvus
|
@@ -121,3 +121,34 @@ To resolve permission issues, you can:
|
|
121
121
|
|
122
122
|
- Review the official Docker documentation on [Windows permission requirements](https://docs.docker.com/desktop/setup/install/windows-permission-requirements/).
|
123
123
|
- Alternatively, follow the community discussion and solutions on [Docker Community Forums](https://forums.docker.com/t/error-when-trying-to-run-windows-containers-docker-client-must-be-run-with-elevated-privileges/136619).
|
124
|
+
|
125
|
+
---
|
126
|
+
|
127
|
+
### Build Images Locally
|
128
|
+
|
129
|
+
If you prefer to build the images yourself:
|
130
|
+
|
131
|
+
```sh
|
132
|
+
git clone https://github.com/VirtualPatientEngine/AIAgents4Pharma.git
|
133
|
+
cd AIAgents4Pharma
|
134
|
+
```
|
135
|
+
|
136
|
+
**GPU build** (overwrites the `latest-gpu` tag locally)
|
137
|
+
|
138
|
+
```sh
|
139
|
+
docker build --platform=linux/amd64 \
|
140
|
+
-t vpatientengine/talk2aiagents4pharma:latest-gpu \
|
141
|
+
-f aiagents4pharma/talk2aiagents4pharma/Dockerfile \
|
142
|
+
--build-arg INSTALL_CUDA=true \
|
143
|
+
.
|
144
|
+
```
|
145
|
+
|
146
|
+
**CPU build** (overwrites the `latest-cpu` tag locally)
|
147
|
+
|
148
|
+
```sh
|
149
|
+
docker build --platform=linux/amd64 \
|
150
|
+
-t vpatientengine/talk2aiagents4pharma:latest-cpu \
|
151
|
+
-f aiagents4pharma/talk2aiagents4pharma/Dockerfile \
|
152
|
+
--build-arg INSTALL_CUDA=false \
|
153
|
+
.
|
154
|
+
```
|
@@ -33,6 +33,7 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
|
|
33
33
|
&& update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1
|
34
34
|
|
35
35
|
FROM python-install AS uv-install
|
36
|
+
ARG INSTALL_CUDA=true
|
36
37
|
WORKDIR /app
|
37
38
|
|
38
39
|
# Install UV package manager and dependencies
|
@@ -42,14 +43,19 @@ RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
|
|
42
43
|
export UV_PROJECT_ENVIRONMENT="/opt/venv" && \
|
43
44
|
uv sync --frozen --extra dev --no-install-project --python python${PYTHON_VERSION} && \
|
44
45
|
. /opt/venv/bin/activate && \
|
45
|
-
|
46
|
-
|
47
|
-
|
48
|
-
|
46
|
+
if [ "$INSTALL_CUDA" = "true" ]; then \
|
47
|
+
uv pip install \
|
48
|
+
--extra-index-url=https://pypi.nvidia.com \
|
49
|
+
--index-strategy unsafe-best-match \
|
50
|
+
cudf-cu12 dask-cudf-cu12; \
|
51
|
+
else \
|
52
|
+
echo "Skipping RAPIDS packages for CPU build"; \
|
53
|
+
fi && \
|
49
54
|
uv cache clean
|
50
55
|
|
51
56
|
FROM ${BASE_IMAGE} AS runtime
|
52
57
|
ARG PYTHON_VERSION=3.12
|
58
|
+
ARG INSTALL_CUDA=true
|
53
59
|
LABEL maintainer="talk2knowledgegraphs"
|
54
60
|
LABEL version="1.0.0"
|
55
61
|
LABEL description="AI Agents for Pharma - Knowledge Graphs Application"
|
@@ -58,6 +64,7 @@ LABEL description="AI Agents for Pharma - Knowledge Graphs Application"
|
|
58
64
|
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
59
65
|
ca-certificates \
|
60
66
|
curl \
|
67
|
+
gnupg \
|
61
68
|
libmagic1 \
|
62
69
|
libopenblas0 \
|
63
70
|
libomp5 \
|
@@ -66,6 +73,27 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
|
|
66
73
|
&& update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
|
67
74
|
&& update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1
|
68
75
|
|
76
|
+
# Install CUDA runtime libraries required by cudf/cupy (optional)
|
77
|
+
RUN if [ "$INSTALL_CUDA" = "true" ]; then \
|
78
|
+
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/3bf863cc.pub \
|
79
|
+
| gpg --dearmor -o /usr/share/keyrings/nvidia-cuda-keyring.gpg && \
|
80
|
+
echo "deb [signed-by=/usr/share/keyrings/nvidia-cuda-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/ /" \
|
81
|
+
> /etc/apt/sources.list.d/nvidia-cuda.list && \
|
82
|
+
apt-get update && \
|
83
|
+
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
84
|
+
cuda-cudart-12-6 \
|
85
|
+
cuda-cudart-dev-12-6 \
|
86
|
+
cuda-nvrtc-12-6 \
|
87
|
+
cuda-nvrtc-dev-12-6 \
|
88
|
+
libcublas-12-6 \
|
89
|
+
libcusparse-12-6 \
|
90
|
+
&& rm -rf /var/lib/apt/lists/*; \
|
91
|
+
else \
|
92
|
+
echo "Skipping CUDA installation"; \
|
93
|
+
fi
|
94
|
+
|
95
|
+
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64
|
96
|
+
|
69
97
|
# Copy UV virtual environment from build stage
|
70
98
|
COPY --from=uv-install /opt/venv /opt/venv
|
71
99
|
|
@@ -11,11 +11,11 @@ LANGCHAIN_API_KEY=your_langchain_api_key_here
|
|
11
11
|
NVIDIA_API_KEY=your_nvidia_api_key_here
|
12
12
|
|
13
13
|
# Set environment variables for data loader
|
14
|
-
MILVUS_HOST=
|
14
|
+
MILVUS_HOST=milvus-standalone
|
15
15
|
MILVUS_PORT=19530
|
16
16
|
MILVUS_USER=root
|
17
17
|
MILVUS_PASSWORD=Milvus
|
18
|
-
MILVUS_DATABASE=
|
18
|
+
MILVUS_DATABASE=t2kg_primekg
|
19
19
|
|
20
20
|
# Specify the data directory for multimodal data to your own data directory
|
21
21
|
# DATA_DIR=/your_absolute_path_to_your_data_dir/
|
@@ -4,7 +4,7 @@ services:
|
|
4
4
|
# talk2knowledgegraphs with automatic data loading via entrypoint
|
5
5
|
talk2knowledgegraphs:
|
6
6
|
container_name: talk2knowledgegraphs
|
7
|
-
image: vpatientengine/talk2knowledgegraphs:latest
|
7
|
+
image: vpatientengine/talk2knowledgegraphs:latest-cpu
|
8
8
|
platform: linux/amd64
|
9
9
|
ports:
|
10
10
|
- "8501:8501"
|
@@ -11,11 +11,11 @@ LANGCHAIN_API_KEY=your_langchain_api_key_here
|
|
11
11
|
NVIDIA_API_KEY=your_nvidia_api_key_here
|
12
12
|
|
13
13
|
# Set environment variables for data loader
|
14
|
-
MILVUS_HOST=
|
14
|
+
MILVUS_HOST=milvus-standalone
|
15
15
|
MILVUS_PORT=19530
|
16
16
|
MILVUS_USER=root
|
17
17
|
MILVUS_PASSWORD=Milvus
|
18
|
-
MILVUS_DATABASE=
|
18
|
+
MILVUS_DATABASE=t2kg_primekg
|
19
19
|
|
20
20
|
# Specify the data directory for multimodal data to your own data directory
|
21
21
|
# DATA_DIR=/your_absolute_path_to_your_data_dir/
|
@@ -4,7 +4,7 @@ services:
|
|
4
4
|
# talk2knowledgegraphs with automatic data loading via entrypoint
|
5
5
|
talk2knowledgegraphs:
|
6
6
|
container_name: talk2knowledgegraphs
|
7
|
-
image: vpatientengine/talk2knowledgegraphs:latest
|
7
|
+
image: vpatientengine/talk2knowledgegraphs:latest-gpu
|
8
8
|
platform: linux/amd64
|
9
9
|
ports:
|
10
10
|
- "8501:8501"
|
@@ -64,7 +64,7 @@ LANGCHAIN_API_KEY=your_langchain_api_key_here
|
|
64
64
|
NVIDIA_API_KEY=your_nvidia_api_key_here
|
65
65
|
|
66
66
|
# Set environment variables for data loader
|
67
|
-
MILVUS_HOST=
|
67
|
+
MILVUS_HOST=milvus-standalone
|
68
68
|
MILVUS_PORT=19530
|
69
69
|
MILVUS_USER=root
|
70
70
|
MILVUS_PASSWORD=Milvus
|
@@ -132,3 +132,34 @@ To resolve permission issues, you can:
|
|
132
132
|
|
133
133
|
- Review the official Docker documentation on [Windows permission requirements](https://docs.docker.com/desktop/setup/install/windows-permission-requirements/).
|
134
134
|
- Alternatively, follow the community discussion and solutions on [Docker Community Forums](https://forums.docker.com/t/error-when-trying-to-run-windows-containers-docker-client-must-be-run-with-elevated-privileges/136619).
|
135
|
+
|
136
|
+
---
|
137
|
+
|
138
|
+
### Build Images Locally
|
139
|
+
|
140
|
+
To build the Docker images yourself:
|
141
|
+
|
142
|
+
```sh
|
143
|
+
git clone https://github.com/VirtualPatientEngine/AIAgents4Pharma.git
|
144
|
+
cd AIAgents4Pharma
|
145
|
+
```
|
146
|
+
|
147
|
+
**GPU build** (overwrites the `latest-gpu` tag locally)
|
148
|
+
|
149
|
+
```sh
|
150
|
+
docker build --platform=linux/amd64 \
|
151
|
+
-t vpatientengine/talk2knowledgegraphs:latest-gpu \
|
152
|
+
-f aiagents4pharma/talk2knowledgegraphs/Dockerfile \
|
153
|
+
--build-arg INSTALL_CUDA=true \
|
154
|
+
.
|
155
|
+
```
|
156
|
+
|
157
|
+
**CPU build** (overwrites the `latest-cpu` tag locally)
|
158
|
+
|
159
|
+
```sh
|
160
|
+
docker build --platform=linux/amd64 \
|
161
|
+
-t vpatientengine/talk2knowledgegraphs:latest-cpu \
|
162
|
+
-f aiagents4pharma/talk2knowledgegraphs/Dockerfile \
|
163
|
+
--build-arg INSTALL_CUDA=false \
|
164
|
+
.
|
165
|
+
```
|
@@ -150,6 +150,7 @@ class DynamicDataLoader:
|
|
150
150
|
self.normalize_vectors = self.use_gpu # Only normalize for GPU (original logic)
|
151
151
|
self.vector_index_type = "GPU_CAGRA" if self.use_gpu else "HNSW"
|
152
152
|
self.metric_type = "IP" if self.use_gpu else "COSINE"
|
153
|
+
self.vector_index_params = self._build_vector_index_params()
|
153
154
|
|
154
155
|
logger.info("Loader Configuration:")
|
155
156
|
logger.info(" Using GPU acceleration: %s", self.use_gpu)
|
@@ -284,6 +285,29 @@ class DynamicDataLoader:
|
|
284
285
|
else:
|
285
286
|
return list(data)
|
286
287
|
|
288
|
+
def _build_vector_index_params(self) -> dict[str, Any]:
|
289
|
+
"""Return index params tuned for the selected backend."""
|
290
|
+
base_params: dict[str, Any] = {
|
291
|
+
"index_type": self.vector_index_type,
|
292
|
+
"metric_type": self.metric_type,
|
293
|
+
}
|
294
|
+
|
295
|
+
if self.vector_index_type == "GPU_CAGRA":
|
296
|
+
base_params["params"] = {
|
297
|
+
"graph_degree": int(os.getenv("CAGRA_GRAPH_DEGREE", "32")),
|
298
|
+
"intermediate_graph_degree": int(
|
299
|
+
os.getenv("CAGRA_INTERMEDIATE_GRAPH_DEGREE", "40")
|
300
|
+
),
|
301
|
+
"search_width": int(os.getenv("CAGRA_SEARCH_WIDTH", "64")),
|
302
|
+
}
|
303
|
+
elif self.vector_index_type == "HNSW":
|
304
|
+
base_params["params"] = {
|
305
|
+
"M": int(os.getenv("HNSW_M", "16")),
|
306
|
+
"efConstruction": int(os.getenv("HNSW_EF_CONSTRUCTION", "200")),
|
307
|
+
}
|
308
|
+
|
309
|
+
return base_params
|
310
|
+
|
287
311
|
def connect_to_milvus(self):
|
288
312
|
"""Connect to Milvus and setup database."""
|
289
313
|
logger.info("Connecting to Milvus at %s:%s", self.milvus_host, self.milvus_port)
|
@@ -442,10 +466,7 @@ class DynamicDataLoader:
|
|
442
466
|
)
|
443
467
|
collection.create_index(
|
444
468
|
field_name="desc_emb",
|
445
|
-
index_params=
|
446
|
-
"index_type": self.vector_index_type,
|
447
|
-
"metric_type": self.metric_type,
|
448
|
-
},
|
469
|
+
index_params=self.vector_index_params.copy(),
|
449
470
|
index_name="desc_emb_index",
|
450
471
|
)
|
451
472
|
|
@@ -569,18 +590,12 @@ class DynamicDataLoader:
|
|
569
590
|
)
|
570
591
|
collection.create_index(
|
571
592
|
field_name="desc_emb",
|
572
|
-
index_params=
|
573
|
-
"index_type": self.vector_index_type,
|
574
|
-
"metric_type": self.metric_type,
|
575
|
-
},
|
593
|
+
index_params=self.vector_index_params.copy(),
|
576
594
|
index_name="desc_emb_index",
|
577
595
|
)
|
578
596
|
collection.create_index(
|
579
597
|
field_name="feat_emb",
|
580
|
-
index_params=
|
581
|
-
"index_type": self.vector_index_type,
|
582
|
-
"metric_type": self.metric_type,
|
583
|
-
},
|
598
|
+
index_params=self.vector_index_params.copy(),
|
584
599
|
index_name="feat_emb_index",
|
585
600
|
)
|
586
601
|
|
@@ -706,10 +721,7 @@ class DynamicDataLoader:
|
|
706
721
|
)
|
707
722
|
collection.create_index(
|
708
723
|
field_name="feat_emb",
|
709
|
-
index_params=
|
710
|
-
"index_type": self.vector_index_type,
|
711
|
-
"metric_type": self.metric_type,
|
712
|
-
},
|
724
|
+
index_params=self.vector_index_params.copy(),
|
713
725
|
index_name="feat_emb_index",
|
714
726
|
)
|
715
727
|
|
@@ -796,8 +808,7 @@ class DynamicDataLoader:
|
|
796
808
|
logger.info(" %s: %d entities", coll, collection.num_entities)
|
797
809
|
|
798
810
|
except Exception:
|
799
|
-
logger.
|
800
|
-
logger.debug("Detailed error information available in debug mode")
|
811
|
+
logger.exception("Error occurred during data loading")
|
801
812
|
raise
|
802
813
|
|
803
814
|
|
@@ -867,8 +878,7 @@ def main():
|
|
867
878
|
logger.info("Data loading interrupted by user")
|
868
879
|
sys.exit(1)
|
869
880
|
except Exception:
|
870
|
-
logger.
|
871
|
-
logger.debug("Detailed error information available in debug mode")
|
881
|
+
logger.exception("Fatal error occurred during data loading")
|
872
882
|
sys.exit(1)
|
873
883
|
|
874
884
|
|
@@ -1,9 +1,9 @@
|
|
1
1
|
aiagents4pharma/__init__.py,sha256=B-tLRCbWgti-jlCnW_qknNKrG4j1t9nhBb-gXaz0Wtg,187
|
2
2
|
aiagents4pharma/talk2aiagents4pharma/.dockerignore,sha256=-hAM7RzkGbjDeU411-kXOmYzNfl3Z9OlLWvN9zMDAXE,89
|
3
|
-
aiagents4pharma/talk2aiagents4pharma/Dockerfile,sha256=
|
3
|
+
aiagents4pharma/talk2aiagents4pharma/Dockerfile,sha256=yozGxCQVzd98dLs7DTLyQB_qfYZtYWtmhSqWsUfWL_Y,4542
|
4
4
|
aiagents4pharma/talk2aiagents4pharma/README.md,sha256=0eGxj7jxi_LrCvX-4I4KrQv-7T2ivo3pqLslG7suaCk,74
|
5
5
|
aiagents4pharma/talk2aiagents4pharma/__init__.py,sha256=gjVTAhBHKPEFBbv_2T-MWuDdwHhAKfWIo-lQSrcsLNE,97
|
6
|
-
aiagents4pharma/talk2aiagents4pharma/install.md,sha256=
|
6
|
+
aiagents4pharma/talk2aiagents4pharma/install.md,sha256=cJPJmg66mT9wCALF1eywKgJb2xKK2r9zbGTCZy6i128,4892
|
7
7
|
aiagents4pharma/talk2aiagents4pharma/agents/__init__.py,sha256=NpNI6Vr9XIr5m0ZaO32c6NEUTDOZvJUqd8gKzNZhcSw,130
|
8
8
|
aiagents4pharma/talk2aiagents4pharma/agents/main_agent.py,sha256=1nRIhj3huv9eVT7v3nhSvx1dDOEEaiApPi7AMRDviXE,2750
|
9
9
|
aiagents4pharma/talk2aiagents4pharma/configs/__init__.py,sha256=hwLAR-uhZGEbD5R7mp4kiltSvxuKkG6-_ac17sF-4xU,68
|
@@ -13,10 +13,10 @@ aiagents4pharma/talk2aiagents4pharma/configs/agents/main_agent/default.yaml,sha2
|
|
13
13
|
aiagents4pharma/talk2aiagents4pharma/configs/app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
14
14
|
aiagents4pharma/talk2aiagents4pharma/configs/app/frontend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
15
15
|
aiagents4pharma/talk2aiagents4pharma/configs/app/frontend/default.yaml,sha256=BpDv_Bau0F7xDPzMHVYxedKpZQAPdo0rAsRnfH32-U8,3470
|
16
|
-
aiagents4pharma/talk2aiagents4pharma/docker-compose/cpu/.env.example,sha256=
|
17
|
-
aiagents4pharma/talk2aiagents4pharma/docker-compose/cpu/docker-compose.yml,sha256=
|
18
|
-
aiagents4pharma/talk2aiagents4pharma/docker-compose/gpu/.env.example,sha256=
|
19
|
-
aiagents4pharma/talk2aiagents4pharma/docker-compose/gpu/docker-compose.yml,sha256=
|
16
|
+
aiagents4pharma/talk2aiagents4pharma/docker-compose/cpu/.env.example,sha256=K6OKI1lz-TDf8PTpwBfYcg_hORc74TNkbkzg4HxSA1Q,582
|
17
|
+
aiagents4pharma/talk2aiagents4pharma/docker-compose/cpu/docker-compose.yml,sha256=_LPT-lKj6mqG9pi-EYCnEylpVllPPHL3uwXZ4NgvDfU,2635
|
18
|
+
aiagents4pharma/talk2aiagents4pharma/docker-compose/gpu/.env.example,sha256=K6OKI1lz-TDf8PTpwBfYcg_hORc74TNkbkzg4HxSA1Q,582
|
19
|
+
aiagents4pharma/talk2aiagents4pharma/docker-compose/gpu/docker-compose.yml,sha256=g6a1GKgz9PTx7Zv441Rf5FMYM5D3FQ6rQKkq9PNEY4I,3005
|
20
20
|
aiagents4pharma/talk2aiagents4pharma/states/__init__.py,sha256=VuVAFmoH8p2erE-mYiaa0uoqTuFynBcXia5Y8lmjI34,109
|
21
21
|
aiagents4pharma/talk2aiagents4pharma/states/state_talk2aiagents4pharma.py,sha256=ahquXi8hg6Bk1ttskl9QaYdHq4rDWa_bK7hu81E75M4,468
|
22
22
|
aiagents4pharma/talk2aiagents4pharma/tests/__init__.py,sha256=U3PsTiUZaUBD1IZanFGkDIOdFieDVJtGKQ5-woYUo8c,45
|
@@ -94,12 +94,12 @@ aiagents4pharma/talk2cells/tools/scp_agent/__init__.py,sha256=DAJp26kugSdVHfFyVR
|
|
94
94
|
aiagents4pharma/talk2cells/tools/scp_agent/display_studies.py,sha256=nQltO147j1cFWUJ9mxg3JlWBLsqFivhJ93g1G7gWZko,602
|
95
95
|
aiagents4pharma/talk2cells/tools/scp_agent/search_studies.py,sha256=xjsgYJ8Bn8RIzIu_bgn8D_2I8wzLOJeu9evT_vF15mM,2647
|
96
96
|
aiagents4pharma/talk2knowledgegraphs/.dockerignore,sha256=-hAM7RzkGbjDeU411-kXOmYzNfl3Z9OlLWvN9zMDAXE,89
|
97
|
-
aiagents4pharma/talk2knowledgegraphs/Dockerfile,sha256=
|
97
|
+
aiagents4pharma/talk2knowledgegraphs/Dockerfile,sha256=UQlJNIh9oR4xDDz1rkmaxcB3pn8LzbhNW78f44oAMEA,4453
|
98
98
|
aiagents4pharma/talk2knowledgegraphs/README.md,sha256=0eGxj7jxi_LrCvX-4I4KrQv-7T2ivo3pqLslG7suaCk,74
|
99
99
|
aiagents4pharma/talk2knowledgegraphs/__init__.py,sha256=ZztaRzRlovSXtVX3i9Rvf84ivIjPn8RMPiYRkbkEJ0E,114
|
100
100
|
aiagents4pharma/talk2knowledgegraphs/entrypoint.sh,sha256=EK_jGau1VuW1uTmFWZcKhLMK9VanC5l3q9axF4ZYgmI,5758
|
101
|
-
aiagents4pharma/talk2knowledgegraphs/install.md,sha256=
|
102
|
-
aiagents4pharma/talk2knowledgegraphs/milvus_data_dump.py,sha256=
|
101
|
+
aiagents4pharma/talk2knowledgegraphs/install.md,sha256=n8G8Zm5EI1l6xX0sm44T-eJFk35VxaxoRFYzwyxPS9w,6239
|
102
|
+
aiagents4pharma/talk2knowledgegraphs/milvus_data_dump.py,sha256=RyPYjtF49DxL7MIWmVUprw0Wpu4HErHOAOQ2_wLUfVI,36095
|
103
103
|
aiagents4pharma/talk2knowledgegraphs/agents/__init__.py,sha256=ugUvVYEdjbZ3y_dogfF5hpQ3lFPFrAvLSydlcpbkGo0,93
|
104
104
|
aiagents4pharma/talk2knowledgegraphs/agents/t2kg_agent.py,sha256=GDeSjJNhAqQWagZOxAWUKqDhzUohHViSsu444W9SzRQ,3240
|
105
105
|
aiagents4pharma/talk2knowledgegraphs/configs/__init__.py,sha256=H-yhTbJ_RXBLe3XSto5x6FmVrgbi7y1WKEfiwmKzLAk,87
|
@@ -129,10 +129,10 @@ aiagents4pharma/talk2knowledgegraphs/datasets/biobridge_primekg.py,sha256=M6NtpM
|
|
129
129
|
aiagents4pharma/talk2knowledgegraphs/datasets/dataset.py,sha256=ls0e15uudIbb4zwHMHxjEovmH145RJ_hPeZni89KSnM,411
|
130
130
|
aiagents4pharma/talk2knowledgegraphs/datasets/primekg.py,sha256=1WHlQCAyKjpBiX3JnIsSohUZe8Yi5pY-VDP4tCugxkg,7709
|
131
131
|
aiagents4pharma/talk2knowledgegraphs/datasets/starkqa_primekg.py,sha256=JtP-jNIFDA3xQbzy5DkK2OHin5NnbWUGa_EJ_1OH6vE,7483
|
132
|
-
aiagents4pharma/talk2knowledgegraphs/docker-compose/cpu/.env.example,sha256=
|
133
|
-
aiagents4pharma/talk2knowledgegraphs/docker-compose/cpu/docker-compose.yml,sha256=
|
134
|
-
aiagents4pharma/talk2knowledgegraphs/docker-compose/gpu/.env.example,sha256=
|
135
|
-
aiagents4pharma/talk2knowledgegraphs/docker-compose/gpu/docker-compose.yml,sha256=
|
132
|
+
aiagents4pharma/talk2knowledgegraphs/docker-compose/cpu/.env.example,sha256=K6OKI1lz-TDf8PTpwBfYcg_hORc74TNkbkzg4HxSA1Q,582
|
133
|
+
aiagents4pharma/talk2knowledgegraphs/docker-compose/cpu/docker-compose.yml,sha256=_TjCAeNcugFnR1FptT8vdTIAbYkvQrQ2cGeN-iU7Dbk,2635
|
134
|
+
aiagents4pharma/talk2knowledgegraphs/docker-compose/gpu/.env.example,sha256=K6OKI1lz-TDf8PTpwBfYcg_hORc74TNkbkzg4HxSA1Q,582
|
135
|
+
aiagents4pharma/talk2knowledgegraphs/docker-compose/gpu/docker-compose.yml,sha256=bLkc20oviAWM5T14_IRy1rvhQFmYkYbhZR5zEbFm3_8,3005
|
136
136
|
aiagents4pharma/talk2knowledgegraphs/states/__init__.py,sha256=9X3zHxvtAJFQ0s0VLZ_-iBn4rMVfZWk5CQiWEKJkr0c,109
|
137
137
|
aiagents4pharma/talk2knowledgegraphs/states/state_talk2knowledgegraphs.py,sha256=QECNXd8IWDh3WlMvePcd2T6G5XqjOI9EkZdWmICcCT0,1088
|
138
138
|
aiagents4pharma/talk2knowledgegraphs/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
@@ -330,7 +330,7 @@ aiagents4pharma/talk2scholars/tools/zotero/utils/review_helper.py,sha256=-q-UuzP
|
|
330
330
|
aiagents4pharma/talk2scholars/tools/zotero/utils/write_helper.py,sha256=K1EatPfC-riGyFmkOAS3ReNBaGPY-znne1KqOnFahkI,7339
|
331
331
|
aiagents4pharma/talk2scholars/tools/zotero/utils/zotero_path.py,sha256=sKkfJu3u4LKSZjfoQRfeqz26IESHRwBtcSDzLMLlJMo,6311
|
332
332
|
aiagents4pharma/talk2scholars/tools/zotero/utils/zotero_pdf_downloader.py,sha256=DBrF5IiF7VRP58hUK8T9LST3lQWLFixLUfnpMSTccoQ,4614
|
333
|
-
aiagents4pharma-1.
|
334
|
-
aiagents4pharma-1.
|
335
|
-
aiagents4pharma-1.
|
336
|
-
aiagents4pharma-1.
|
333
|
+
aiagents4pharma-1.48.1.dist-info/METADATA,sha256=IBeKfbyj8tGGv-KJeW26nrzfoOY7O2ptrPM1GpqxS0A,17035
|
334
|
+
aiagents4pharma-1.48.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
335
|
+
aiagents4pharma-1.48.1.dist-info/licenses/LICENSE,sha256=IcIbyB1Hyk5ZDah03VNQvJkbNk2hkBCDqQ8qtnCvB4Q,1077
|
336
|
+
aiagents4pharma-1.48.1.dist-info/RECORD,,
|
File without changes
|
File without changes
|