optimize code and doc

cpc1986 · Sep 10, 2023 · 4153d69 · 4153d69
1 parent 3480728
commit 4153d69
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -31,7 +31,7 @@ The following example can view in the [langchain-example](langchain-examples/src
 ### 3.1 LLMs
 - [OpenAI](langchain-examples/src/main/java/com/hw/langchain/examples/llms/OpenAIExample.java)
 - [Azure OpenAI](openai-client/src/test/java/com/hw/openai/AzureOpenAiClientTest.java)
-- [ChatGLM2-6B](langchain-examples/src/main/java/com/hw/langchain/examples/llms/ChatGLMExample.java)
+- [ChatGLM2](langchain-examples/src/main/java/com/hw/langchain/examples/llms/ChatGLMExample.java)
 - [Ollama](langchain-examples/src/main/java/com/hw/langchain/examples/llms/OllamaExample.java)
 
 ### 3.2 Vector stores

diff --git a/langchain-core/src/main/java/com/hw/langchain/memory/chat/memory/BaseChatMemory.java b/langchain-core/src/main/java/com/hw/langchain/memory/chat/memory/BaseChatMemory.java
@@ -32,7 +32,7 @@
 /**
  * @author HamaWhite
  */
-public abstract class BaseChatMemory extends BaseMemory {
+public abstract class BaseChatMemory implements BaseMemory {
 
     protected BaseChatMessageHistory chatMemory = new ChatMessageHistory();
 

diff --git a/langchain-core/src/main/java/com/hw/langchain/schema/BaseMemory.java b/langchain-core/src/main/java/com/hw/langchain/schema/BaseMemory.java
@@ -26,26 +26,26 @@
  *
  * @author HamaWhite
  */
-public abstract class BaseMemory {
+public interface BaseMemory {
 
     /**
      * Input keys this memory class will load dynamically
      */
-    public abstract List<String> memoryVariables();
+    List<String> memoryVariables();
 
     /**
      * Return key-value pairs given the text input to the chain.
      * If None, return all memories
      */
-    public abstract Map<String, Object> loadMemoryVariables(Map<String, Object> inputs);
+    Map<String, Object> loadMemoryVariables(Map<String, Object> inputs);
 
     /**
      * Save the context of this model run to memory.
      */
-    public abstract void saveContext(Map<String, Object> inputs, Map<String, String> outputs);
+    void saveContext(Map<String, Object> inputs, Map<String, String> outputs);
 
     /**
      * Clear memory contents.
      */
-    public abstract void clear();
+    void clear();
 }