wanjia9506
/
langchain4j
mirror of https://github.com/langchain4j/langchain4j.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixed failing tests

This commit is contained in:
LangChain4j 2024-08-23 15:38:06 +02:00
parent 98f1bbedae
commit 20a30eb253
2 changed files with 34 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
langchain4j-neo4j/pom.xml
View File

@ -111,6 +111,17 @@
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.tinylog</groupId>
<artifactId>tinylog-impl</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.tinylog</groupId>
<artifactId>slf4j-tinylog</artifactId>
<scope>test</scope>
</dependency>
</dependencies>
<build>

45
langchain4j-neo4j/src/test/java/dev/langchain4j/rag/content/retriever/neo4j/Neo4jContentRetrieverIT.java
View File

@ -15,10 +15,10 @@ import org.neo4j.driver.GraphDatabase;
import org.neo4j.driver.Session;
import org.testcontainers.containers.Neo4jContainer;
import org.testcontainers.junit.jupiter.Container;
import org.testcontainers.utility.DockerImageName;
import java.util.List;
import static dev.langchain4j.model.openai.OpenAiChatModelName.GPT_4_O_MINI;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.when;
@ -27,41 +27,35 @@ import static org.mockito.Mockito.when;
class Neo4jContentRetrieverIT {
@Container
private static final Neo4jContainer<?> neo4jContainer = new Neo4jContainer<>(DockerImageName.parse("neo4j:5.16.0"))
private static final Neo4jContainer<?> neo4jContainer = new Neo4jContainer<>("neo4j:latest")
.withoutAuthentication()
.withLabsPlugins("apoc");
.withPlugins("apoc");
private static Driver driver;
private static Neo4jGraph graph;
private static Neo4jContentRetriever retriever;
private Driver driver;
private Neo4jGraph graph;
private Neo4jContentRetriever retriever;
@Mock
private ChatLanguageModel chatLanguageModel;
@BeforeAll
static void beforeAll() {
neo4jContainer.start();
driver = GraphDatabase.driver(neo4jContainer.getBoltUrl(), AuthTokens.none());
graph = Neo4jGraph.builder().driver(driver).build();
}
@AfterAll
static void afterAll() {
graph.close();
driver.close();
neo4jContainer.stop();
}
@BeforeEach
void setUp() {
void beforeEach() {
driver = GraphDatabase.driver(neo4jContainer.getBoltUrl(), AuthTokens.none());
try (Session session = driver.session()) {
session.run("CREATE (book:Book {title: 'Dune'})<-[:WROTE]-(author:Person {name: 'Frank Herbert'})");
}
graph = Neo4jGraph.builder()
.driver(driver)
.build();
retriever = Neo4jContentRetriever.builder()
.graph(graph)
.chatLanguageModel(chatLanguageModel)
@ -69,11 +63,17 @@ class Neo4jContentRetrieverIT {
}
@AfterEach
void tearDown() {
void afterEach() {
try (Session session = driver.session()) {
session.run("MATCH (n) DETACH DELETE n");
}
graph.close();
driver.close();
}
@AfterAll
static void afterAll() {
neo4jContainer.stop();
}
@Test
@ -110,6 +110,7 @@ class Neo4jContentRetrieverIT {
.baseUrl(System.getenv("OPENAI_BASE_URL"))
.apiKey(System.getenv("OPENAI_API_KEY"))
.organizationId(System.getenv("OPENAI_ORGANIZATION_ID"))
.modelName(GPT_4_O_MINI)
.logRequests(true)
.logResponses(true)
.build();