LangChain for Java: Supercharge your Java application with the power of LLMs

Project goals

The goal of this project is to simplify the integration of AI/LLM capabilities into your Java application.

This can be achieved thanks to:

• A simple and coherent layer of abstractions, designed to ensure that your code does not depend on concrete implementations such as LLM providers, embedding store providers, etc. This allows for easy swapping of components.
• Numerous implementations of the above-mentioned abstractions, providing you with a variety of LLMs and embedding stores to choose from.
• Range of in-demand features on top of LLMs, such as:
  • The capability to ingest your own data (documentation, codebase, etc.), allowing the LLM to act and respond based on your data.
  • Autonomous agents for delegating tasks (defined on the fly) to the LLM, which will strive to complete them.
  • Prompt templates to help you achieve the highest possible quality of LLM responses.
  • Memory to provide context to the LLM for your current and past conversations.
  • Structured outputs for receiving responses from the LLM with a desired structure as Java POJOs.
  • "AI Services" for declaratively defining complex AI behavior behind a simple API.
  • Chains to reduce the need for extensive boilerplate code in common use-cases.
  • Auto-moderation to ensure that all inputs and outputs to/from the LLM are not harmful.

Code examples

Please see examples of how LangChain4j can be used in langchain4j-examples repo:

• Examples in plain Java
• Example with Spring Boot

Quarkus specific examples (leveraging the quarkus-langchain4j dependency which builds on this project) can be found here

News

12 November:

• Integration with OpenSearch by @riferrei
• Add support for loading documents from S3 by @jmgang
• Integration with PGVector by @kevin-wu-os
• Integration with Ollama by @Martin7-1
• Integration with Amazon Bedrock by @pascalconfluent
• Adding Memory Id to Tool Method Call by @benedictstrube
• And more

29 September:

• Updates to models API: return Response<T> instead of T. Response<T> contains token usage and finish reason.
• All model and embedding store integrations now live in their own modules
• Integration with Vespa by @Heezer
• Integration with Elasticsearch by @Martin7-1
• Integration with Redis by @Martin7-1
• Integration with Milvus by @IuriiKoval
• Integration with Astra DB and Cassandra by @clun
• Added support for overlap in document splitters
• Some bugfixes and smaller improvements

29 August:

• Offline text classification with embeddings
• Integration with Google Vertex AI by @kuraleta
• Reworked document splitters
• In-memory embedding store can now be easily persisted
• And more

19 August:

• Integration with Azure OpenAI by @kuraleta
• Integration with Qwen models (DashScope) by @jiangsier-xyz
• Integration with Chroma by @kuraleta
• Support for persistent ChatMemory

10 August:

• Integration with Weaviate by @Heezer
• Support for DOC, XLS and PPT document types by @oognuyh
• Separate chat memory for each user
• Custom in-process embedding models
• Added lots of Javadoc
• And more

26 July:

• We've added integration with LocalAI. Now, you can use LLMs hosted locally!
• Added support for response streaming in AI Services.

21 July:

• Now, you can do text embedding inside your JVM.

17 July:

• You can now try out OpenAI's gpt-3.5-turbo and text-embedding-ada-002 models with LangChain4j for free, without needing an OpenAI account and keys! Simply use the API key "demo".

15 July:

• Added EmbeddingStoreIngestor
• Redesigned document loaders (see FileSystemDocumentLoader)
• Simplified ConversationalRetrievalChain
• Renamed DocumentSegment into TextSegment
• Added output parsers for numeric types
• Added @UserName for AI Services
• Fixed 23 and 24

11 July:

• Added "Dynamic Tools": Now, the LLM can generate code for tasks that require precise calculations, such as math and string manipulation. This will be dynamically executed in a style akin to GPT-4's code interpreter! We use Judge0, hosted by Rapid API, for code execution. You can subscribe and receive 50 free executions per day.

5 July:

• Now you can add your custom knowledge base to "AI Services". Relevant information will be automatically retrieved and injected into the prompt. This way, the LLM will have a context of your data and will answer based on it!
• The current date and time can now be automatically injected into the prompt using special {{current_date}}, {{current_time}} and {{current_date_time}} placeholders.

3 July:

• Added support for Spring Boot 3

2 July:

• Added Spring Boot Starter
• Added support for HuggingFace models

1 July:

• Added "Tools" (support for OpenAI functions)

Highlights

You can declaratively define concise "AI Services" that are powered by LLMs:

interface Assistant {

    String chat(String userMessage);
}

Assistant assistant = AiServices.create(Assistant.class, model);

String answer = assistant.chat("Hello");
    
System.out.println(answer);
// Hello! How can I assist you today?

You can use LLM as a classifier:

enum Sentiment {
    POSITIVE, NEUTRAL, NEGATIVE
}

interface SentimentAnalyzer {

    @UserMessage("Analyze sentiment of {{it}}")
    Sentiment analyzeSentimentOf(String text);

    @UserMessage("Does {{it}} have a positive sentiment?")
    boolean isPositive(String text);
}

SentimentAnalyzer sentimentAnalyzer = AiServices.create(SentimentAnalyzer.class, model);

Sentiment sentiment = sentimentAnalyzer.analyzeSentimentOf("It is good!");
// POSITIVE

boolean positive = sentimentAnalyzer.isPositive("It is bad!");
// false

You can easily extract structured information from unstructured data:

class Person {

    private String firstName;
    private String lastName;
    private LocalDate birthDate;

    public String toString() {...}
}

interface PersonExtractor {

    @UserMessage("Extract information about a person from {{it}}")
    Person extractPersonFrom(String text);
}

PersonExtractor extractor = AiServices.create(PersonExtractor.class, model);

String text = "In 1968, amidst the fading echoes of Independence Day, "
    + "a child named John arrived under the calm evening sky. "
    + "This newborn, bearing the surname Doe, marked the start of a new journey.";

Person person = extractor.extractPersonFrom(text);
// Person { firstName = "John", lastName = "Doe", birthDate = 1968-07-04 }

You can define more sophisticated prompt templates using mustache syntax:

interface Translator {

    @SystemMessage("You are a professional translator into {{language}}")
    @UserMessage("Translate the following text: {{text}}")
    String translate(@V("text") String text, @V("language") String language);
}

Translator translator = AiServices.create(Translator.class, model);

String translation = translator.translate("Hello, how are you?", "Italian");
// Ciao, come stai?

You can provide tools that LLMs can use! Can be anything: retrieve information from DB, call APIs, etc. See example here.

Compatibility

• Java: 8 or higher
• Spring Boot: 2 or 3

Getting started

1. Add LangChain4j OpenAI dependency to your project:

   • Maven:

     <dependency>
         <groupId>dev.langchain4j</groupId>
         <artifactId>langchain4j-open-ai</artifactId>
         <version>0.24.0</version>
     </dependency>
     

   • Gradle:

     implementation 'dev.langchain4j:langchain4j-open-ai:0.24.0'
     

2. Import your OpenAI API key:

   String apiKey = System.getenv("OPENAI_API_KEY");

   You can use the API key "demo" to test OpenAI, which we provide for free. How to gen an API key?

3. Create an instance of a model and start interacting:

   OpenAiChatModel model = OpenAiChatModel.withApiKey(apiKey);
   
   String answer = model.generate("Hello world!");
   
   System.out.println(answer); // Hello! How can I assist you today?

Disclaimer

Please note that the library is in active development and:

• Many features are still missing. We are working hard on implementing them ASAP.
• API might change at any moment. At this point, we prioritize good design in the future over backward compatibility now. We hope for your understanding.
• We need your input! Please let us know what features you need and your concerns about the current implementation.

Current capabilities:

• AI Services:
  • Simple
  • With Memory
  • With Tools
  • With Streaming
  • With Retriever
  • With Auto-Moderation
  • With Structured Outputs, Structured Prompts, etc
• Integration with OpenAI and Azure OpenAI for:
  • Chats (sync + streaming + functions)
  • Completions (sync + streaming)
  • Embeddings
• Integration with Google Vertex AI for:
  • Chats
  • Completions
  • Embeddings
• Integration with HuggingFace Inference API for:
  • Chats
  • Completions
  • Embeddings
• Integration with LocalAI for:
  • Chats (sync + streaming + functions)
  • Completions (sync + streaming)
  • Embeddings
• Integration with DashScope for:
  • Chats (sync + streaming)
  • Completions (sync + streaming)
  • Embeddings
• Chat memory
• Persistent chat memory
• Chat with Documents
• Integration with Astra DB and Cassandra
• Integration with Chroma
• Integration with Elasticsearch
• Integration with Milvus
• Integration with Pinecone
• Integration with Redis
• Integration with Vespa
• Integration with Weaviate
• In-memory embedding store (can be persisted)
• Structured outputs
• Prompt templates
• Structured prompt templates
• Streaming of LLM responses
• Loading txt, html, pdf, doc, xls and ppt documents from the file system and via URL
• Splitting documents into segments:
  • by paragraphs, lines, sentences, words, etc
  • recursively
  • with overlap
• Token count estimation (so that you can predict how much you will pay)

Coming soon:

• Extending "AI Service" features
• Integration with more LLM providers (commercial and free)
• Integrations with more embedding stores (commercial and free)
• Support for more document types
• Long-term memory for chatbots and agents
• Chain-of-Thought and Tree-of-Thought

Request features

Please let us know what features you need!

Contribute

Please help us make this open-source library better by contributing.

Some guidelines:

1. Follow Google's Best Practices for Java Libraries.
2. Keep the code compatible with Java 8.
3. Avoid adding new dependencies as much as possible. If absolutely necessary, try to (re)use the same libraries which are already present.
4. Follow existing code styles present in the project.
5. Ensure to add Javadoc where necessary.
6. Provide unit and/or integration tests for your code.
7. Large features should be discussed with maintainers before implementation.

Use cases

You might ask why would I need all of this? Here are a couple of examples:

• You want to implement a custom AI-powered chatbot that has access to your data and behaves the way you want it:
  • Customer support chatbot that can:
    • politely answer customer questions
    • take /change/cancel orders
  • Educational assistant that can:
    • Teach various subjects
    • Explain unclear parts
    • Assess user's understanding/knowledge
• You want to process a lot of unstructured data (files, web pages, etc) and extract structured information from them. For example:
  • extract insights from customer reviews and support chat history
  • extract interesting information from the websites of your competitors
  • extract insights from CVs of job applicants
• You want to generate information, for example:
  • Emails tailored for each of your customers
  • Content for your app/website:
    • Blog posts
    • Stories
• You want to transform information, for example:
  • Summarize
  • Proofread and rewrite
  • Translate

Best practices

We highly recommend watching this amazing 90-minute tutorial on prompt engineering best practices, presented by Andrew Ng (DeepLearning.AI) and Isa Fulford (OpenAI). This course will teach you how to use LLMs efficiently and achieve the best possible results. Good investment of your time!

Here are some best practices for using LLMs:

• Be responsible. Use AI for Good.
• Be specific. The more specific your query, the best results you will get.
• Add a "Let’s think step by step" instruction to your prompt.
• Specify steps to achieve the desired goal yourself. This will make the LLM do what you want it to do.
• Provide examples. Sometimes it is best to show LLM a few examples of what you want instead of trying to explain it.
• Ask LLM to provide structured output (JSON, XML, etc). This way you can parse response more easily and distinguish different parts of it.
• Use unusual delimiters, such as ```triple backticks``` to help the LLM distinguish data or input from instructions.

How to get an API key

You will need an API key from OpenAI (paid) or HuggingFace (free) to use LLMs hosted by them.

We recommend using OpenAI LLMs (gpt-3.5-turbo and gpt-4) as they are by far the most capable and are reasonably priced.

It will cost approximately $0.01 to generate 10 pages (A4 format) of text with gpt-3.5-turbo. With gpt-4, the cost will be $0.30 to generate the same amount of text. However, for some use cases, this higher cost may be justified.

How to get OpenAI API key.

For embeddings, we recommend using one of the models from the HuggingFace MTEB leaderboard. You'll have to find the best one for your specific use case.

Here's how to get a HuggingFace API key:

• Create an account on https://huggingface.co
• Go to https://huggingface.co/settings/tokens
• Generate a new access token