Skip to content

Commit

Permalink
Setup docker compose to use host networking, so it can access qdrant …
Browse files Browse the repository at this point in the history
…and redis servers via localhost
  • Loading branch information
philip06 committed May 14, 2023
1 parent 21983c5 commit 6e71366
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 12 deletions.
5 changes: 0 additions & 5 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,3 @@ FROM rust

# install tini to capture SIGINTs properly
RUN apt-get update && apt-get install -y tini

WORKDIR /smartgpt

ENTRYPOINT ["/usr/bin/tini", "--"]
CMD ["cargo", "run", "--release"]
19 changes: 15 additions & 4 deletions docker-compose.yaml
Original file line number Diff line number Diff line change
@@ -1,22 +1,33 @@
services:
smartgpt:
build: .
working_dir: /smartgpt
network_mode: host
# depends_on:
# - qdrant
# - redis
ports:
- "3001:3001"
volumes:
- .:/smartgpt
environment:
QDRANT_HOST: ${QDRANT_HOST:-qdrant}
REDIS_HOST: ${REDIS_HOST:-redis}
# environment:
# QDRANT_HOST: ${QDRANT_HOST:-qdrant}
# REDIS_HOST: ${REDIS_HOST:-redis}
entrypoint:
- /usr/bin/tini
- --
command:
- cargo
- run
- --release
qdrant:
image: "qdrant/qdrant"
network_mode: host
ports:
- "6333:6333"
- "6334:6334"
environment:
QDRANT__SERVICE__GRPC_PORT: 6334
redis:
image: "redis/redis-stack-server:latest"
image: "redis/redis-stack-server:latest"
network_mode: host
6 changes: 3 additions & 3 deletions src/memory/qdrant.rs
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ use qdrant_client::prelude::*;
use qdrant_client::qdrant::value::Kind;
use qdrant_client::qdrant::vectors::VectorsOptions;
use qdrant_client::qdrant::vectors_config::Config;
use qdrant_client::qdrant::{CreateCollection, SearchPoints, VectorParams, VectorsConfig, PointId, Vectors, Vector, WithPayloadSelector, with_payload_selector, OptimizersConfigDiff, WalConfigDiff, HnswConfigDiff, QuantizationConfig, quantization_config, ScalarQuantization, RecommendPoints, ScoredPoint};
use qdrant_client::qdrant::{CreateCollection, SearchPoints, VectorParams, VectorsConfig, PointId, Vectors, Vector, WithPayloadSelector, with_payload_selector, RecommendPoints, ScoredPoint};
use tokio::runtime::Runtime;

use super::MemorySystem;
Expand Down Expand Up @@ -97,7 +97,7 @@ impl MemorySystem for QdrantMemorySystem {
let latest_point_id = latest_point_id_option.unwrap_or(0);

let mut points: Vec<PointId> = vec![];
let mut search_result = vec![];
let search_result;
if latest_point_id > 0 {
points.push(PointId {
point_id_options: Some(point_id::PointIdOptions::Num(latest_point_id)),
Expand Down Expand Up @@ -174,7 +174,7 @@ impl MemoryProvider for QdrantProvider {
init_qdrant_client().await
})?;

let collection_name = "qdrant_memory";
let collection_name = "smartgpt_agent_memory";

rt.block_on(async {
create_collection_if_not_exists(&client, &collection_name).await
Expand Down

0 comments on commit 6e71366

Please sign in to comment.