mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-08-28 05:25:04 +00:00
rename experiencemaker
This commit is contained in:
parent
205d4e4776
commit
195d2aef4d
62 changed files with 94 additions and 94 deletions
|
|
@ -1,7 +1,7 @@
|
|||
from experiencescope.model.openai_compatible_embedding_model import OpenAICompatibleEmbeddingModel
|
||||
from experiencescope.model.openai_compatible_llm import OpenAICompatibleBaseLLM
|
||||
from experiencemaker.model.openai_compatible_embedding_model import OpenAICompatibleEmbeddingModel
|
||||
from experiencemaker.model.openai_compatible_llm import OpenAICompatibleBaseLLM
|
||||
|
||||
from experiencescope.utils.registry import Registry
|
||||
from experiencemaker.utils.registry import Registry
|
||||
|
||||
LLM_REGISTRY = Registry("llm")
|
||||
LLM_REGISTRY.register(OpenAICompatibleBaseLLM, "openai_compatible")
|
||||
|
|
@ -4,7 +4,7 @@ from typing import List
|
|||
from loguru import logger
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from experiencescope.schema.vector_store_node import VectorStoreNode
|
||||
from experiencemaker.schema.vector_store_node import VectorStoreNode
|
||||
|
||||
|
||||
class BaseEmbeddingModel(BaseModel, ABC):
|
||||
|
|
@ -4,8 +4,8 @@ from typing import List, Literal
|
|||
from loguru import logger
|
||||
from pydantic import Field, BaseModel
|
||||
|
||||
from experiencescope.schema.trajectory import Message, ActionMessage
|
||||
from experiencescope.tool.base_tool import BaseTool
|
||||
from experiencemaker.schema.trajectory import Message, ActionMessage
|
||||
from experiencemaker.tool.base_tool import BaseTool
|
||||
|
||||
|
||||
class BaseLLM(BaseModel, ABC):
|
||||
|
|
@ -4,7 +4,7 @@ from typing import Literal, List
|
|||
from openai import OpenAI
|
||||
from pydantic import Field, PrivateAttr, model_validator
|
||||
|
||||
from experiencescope.model.base_embedding_model import BaseEmbeddingModel
|
||||
from experiencemaker.model.base_embedding_model import BaseEmbeddingModel
|
||||
|
||||
|
||||
class OpenAICompatibleEmbeddingModel(BaseEmbeddingModel):
|
||||
|
|
@ -6,10 +6,10 @@ from openai import OpenAI
|
|||
from openai.types import CompletionUsage
|
||||
from pydantic import Field, PrivateAttr, model_validator
|
||||
|
||||
from experiencescope.enumeration.chunk_enum import ChunkEnum
|
||||
from experiencescope.model.base_llm import BaseLLM
|
||||
from experiencescope.schema.trajectory import Message, ActionMessage, ToolCall
|
||||
from experiencescope.tool.base_tool import BaseTool
|
||||
from experiencemaker.enumeration.chunk_enum import ChunkEnum
|
||||
from experiencemaker.model.base_llm import BaseLLM
|
||||
from experiencemaker.schema.trajectory import Message, ActionMessage, ToolCall
|
||||
from experiencemaker.tool.base_tool import BaseTool
|
||||
|
||||
|
||||
class OpenAICompatibleBaseLLM(BaseLLM):
|
||||
|
|
@ -4,9 +4,9 @@ from typing import List
|
|||
from loguru import logger
|
||||
from pydantic import Field
|
||||
|
||||
from experiencescope.module.agent_wrapper.base_agent_wrapper_mixin import BaseAgentWrapperMixin
|
||||
from experiencescope.module.environment.base_environment import BaseEnvironment
|
||||
from experiencescope.schema.trajectory import Trajectory, Message, StateMessage, ActionMessage, ContextMessage
|
||||
from experiencemaker.module.agent_wrapper.base_agent_wrapper_mixin import BaseAgentWrapperMixin
|
||||
from experiencemaker.module.environment.base_environment import BaseEnvironment
|
||||
from experiencemaker.schema.trajectory import Trajectory, Message, StateMessage, ActionMessage, ContextMessage
|
||||
|
||||
|
||||
class BaseAgentWrapper(BaseAgentWrapperMixin, ABC):
|
||||
|
|
@ -2,9 +2,9 @@ from abc import ABC
|
|||
|
||||
from pydantic import Field
|
||||
|
||||
from experiencescope.module.base_module import BaseModule
|
||||
from experiencescope.module.context_generator.base_context_generator import BaseContextGenerator
|
||||
from experiencescope.schema.trajectory import Trajectory, ActionMessage, Message
|
||||
from experiencemaker.module.base_module import BaseModule
|
||||
from experiencemaker.module.context_generator.base_context_generator import BaseContextGenerator
|
||||
from experiencemaker.schema.trajectory import Trajectory, ActionMessage, Message
|
||||
|
||||
|
||||
class BaseAgentWrapperMixin(BaseModule, ABC):
|
||||
|
|
@ -1,10 +1,10 @@
|
|||
import datetime
|
||||
|
||||
from experiencescope.module.agent_wrapper.base_agent_wrapper_v2 import BaseAgentWrapperV2
|
||||
from experiencemaker.module.agent_wrapper.base_agent_wrapper_v2 import BaseAgentWrapperV2
|
||||
from loguru import logger
|
||||
|
||||
from experiencescope.module.environment.base_environment import BaseEnvironment
|
||||
from experiencescope.schema.trajectory import Message, StateMessage, ContextMessage, ActionMessage
|
||||
from experiencemaker.module.environment.base_environment import BaseEnvironment
|
||||
from experiencemaker.schema.trajectory import Message, StateMessage, ContextMessage, ActionMessage
|
||||
|
||||
|
||||
class NaiveAgentWrapper(BaseAgentWrapperV2):
|
||||
|
|
@ -3,10 +3,10 @@ from abc import ABC
|
|||
from loguru import logger
|
||||
from pydantic import BaseModel, Field, model_validator
|
||||
|
||||
from experiencescope.model import LLM_REGISTRY, EMBEDDING_MODEL_REGISTRY
|
||||
from experiencescope.model.base_embedding_model import BaseEmbeddingModel
|
||||
from experiencescope.model.base_llm import BaseLLM
|
||||
from experiencescope.utils.prompt_handler import PromptHandler
|
||||
from experiencemaker.model import LLM_REGISTRY, EMBEDDING_MODEL_REGISTRY
|
||||
from experiencemaker.model.base_embedding_model import BaseEmbeddingModel
|
||||
from experiencemaker.model.base_llm import BaseLLM
|
||||
from experiencemaker.utils.prompt_handler import PromptHandler
|
||||
|
||||
|
||||
class BaseModule(BaseModel, ABC):
|
||||
|
|
@ -3,10 +3,10 @@ from typing import List
|
|||
|
||||
from pydantic import Field
|
||||
|
||||
from experiencescope.module.base_module import BaseModule
|
||||
from experiencescope.schema.trajectory import Trajectory, ContextMessage
|
||||
from experiencescope.schema.vector_store_node import VectorStoreNode
|
||||
from experiencescope.storage.base_vector_store import BaseVectorStore
|
||||
from experiencemaker.module.base_module import BaseModule
|
||||
from experiencemaker.schema.trajectory import Trajectory, ContextMessage
|
||||
from experiencemaker.schema.vector_store_node import VectorStoreNode
|
||||
from experiencemaker.storage.base_vector_store import BaseVectorStore
|
||||
|
||||
|
||||
class BaseContextGenerator(BaseModule, ABC):
|
||||
|
|
@ -2,11 +2,11 @@ from typing import List
|
|||
|
||||
from pydantic import Field
|
||||
|
||||
from experiencescope.module.base_module import BaseModule
|
||||
from experiencescope.module.reward_fn.base_reward_fn import BaseRewardFn
|
||||
from experiencescope.schema.reward import Reward
|
||||
from experiencescope.schema.trajectory import StateMessage, ActionMessage, ToolCall
|
||||
from experiencescope.tool.base_tool import BaseTool
|
||||
from experiencemaker.module.base_module import BaseModule
|
||||
from experiencemaker.module.reward_fn.base_reward_fn import BaseRewardFn
|
||||
from experiencemaker.schema.reward import Reward
|
||||
from experiencemaker.schema.trajectory import StateMessage, ActionMessage, ToolCall
|
||||
from experiencemaker.tool.base_tool import BaseTool
|
||||
|
||||
|
||||
class BaseEnvironment(BaseModule):
|
||||
|
|
@ -2,10 +2,10 @@ from abc import ABC
|
|||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from experiencescope.module.agent_wrapper.base_agent_wrapper import BaseAgentWrapper
|
||||
from experiencescope.module.context_generator.base_context_generator import BaseContextGenerator
|
||||
from experiencescope.module.environment.base_environment import BaseEnvironment
|
||||
from experiencescope.module.summarizer.base_summarizer import BaseSummarizer
|
||||
from experiencemaker.module.agent_wrapper.base_agent_wrapper import BaseAgentWrapper
|
||||
from experiencemaker.module.context_generator.base_context_generator import BaseContextGenerator
|
||||
from experiencemaker.module.environment.base_environment import BaseEnvironment
|
||||
from experiencemaker.module.summarizer.base_summarizer import BaseSummarizer
|
||||
|
||||
|
||||
class BaseEvaluator(BaseModel, ABC):
|
||||
|
|
@ -1,8 +1,8 @@
|
|||
from abc import ABC
|
||||
|
||||
from experiencescope.module.base_module import BaseModule
|
||||
from experiencescope.schema.reward import Reward
|
||||
from experiencescope.schema.trajectory import Trajectory
|
||||
from experiencemaker.module.base_module import BaseModule
|
||||
from experiencemaker.schema.reward import Reward
|
||||
from experiencemaker.schema.trajectory import Trajectory
|
||||
|
||||
|
||||
class BaseRewardFn(BaseModule, ABC):
|
||||
|
|
@ -2,11 +2,11 @@ from typing import List
|
|||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from experiencescope.module.agent_wrapper.base_agent_wrapper import BaseAgentWrapper
|
||||
from experiencescope.module.context_generator.base_context_generator import BaseContextGenerator
|
||||
from experiencescope.module.environment.base_environment import BaseEnvironment
|
||||
from experiencescope.module.summarizer.base_summarizer import BaseSummarizer
|
||||
from experiencescope.schema.trajectory import Trajectory
|
||||
from experiencemaker.module.agent_wrapper.base_agent_wrapper import BaseAgentWrapper
|
||||
from experiencemaker.module.context_generator.base_context_generator import BaseContextGenerator
|
||||
from experiencemaker.module.environment.base_environment import BaseEnvironment
|
||||
from experiencemaker.module.summarizer.base_summarizer import BaseSummarizer
|
||||
from experiencemaker.schema.trajectory import Trajectory
|
||||
|
||||
|
||||
class BaseRunner(BaseModel):
|
||||
|
|
@ -2,9 +2,9 @@ from typing import List
|
|||
|
||||
from pydantic import Field
|
||||
|
||||
from experiencescope.module.base_module import BaseModule
|
||||
from experiencescope.schema.trajectory import Trajectory, Sample, SummaryMessage
|
||||
from experiencescope.storage.base_vector_store import BaseVectorStore
|
||||
from experiencemaker.module.base_module import BaseModule
|
||||
from experiencemaker.schema.trajectory import Trajectory, Sample, SummaryMessage
|
||||
from experiencemaker.storage.base_vector_store import BaseVectorStore
|
||||
|
||||
|
||||
class BaseSummarizer(BaseModule):
|
||||
|
|
@ -3,7 +3,7 @@ from typing import List
|
|||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from experiencescope.schema.trajectory import Trajectory
|
||||
from experiencemaker.schema.trajectory import Trajectory
|
||||
|
||||
|
||||
class BaseTrainner(BaseModel, ABC):
|
||||
|
|
@ -2,7 +2,7 @@ from importlib import import_module
|
|||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from experiencescope.utils.file_handler import FileHandler
|
||||
from experiencemaker.utils.file_handler import FileHandler
|
||||
|
||||
|
||||
class ModuleLoader(BaseModel):
|
||||
|
|
@ -3,8 +3,8 @@ from typing import List
|
|||
|
||||
from pydantic import Field
|
||||
|
||||
from experiencescope.schema.module_loader import ModuleLoader
|
||||
from experiencescope.schema.trajectory import Trajectory
|
||||
from experiencemaker.schema.module_loader import ModuleLoader
|
||||
from experiencemaker.schema.trajectory import Trajectory
|
||||
|
||||
|
||||
class BaseRequest(ModuleLoader, ABC):
|
||||
|
|
@ -3,7 +3,7 @@ from typing import List
|
|||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from experiencescope.schema.trajectory import Trajectory, ContextMessage, Sample
|
||||
from experiencemaker.schema.trajectory import Trajectory, ContextMessage, Sample
|
||||
|
||||
|
||||
class BaseResponse(BaseModel, ABC):
|
||||
|
|
@ -5,7 +5,7 @@ from uuid import uuid4
|
|||
|
||||
from pydantic import BaseModel, Field, model_validator
|
||||
|
||||
from experiencescope.enumeration.role import Role
|
||||
from experiencemaker.enumeration.role import Role
|
||||
|
||||
|
||||
class ToolCall(BaseModel):
|
||||
|
|
@ -1,14 +1,14 @@
|
|||
from experiencescope.utils.logger import init_logger
|
||||
from experiencemaker.utils.logger import init_logger
|
||||
init_logger()
|
||||
|
||||
from typing import List
|
||||
from fastapi import FastAPI
|
||||
from experiencescope.module.agent_wrapper.base_agent_wrapper import BaseAgentWrapper
|
||||
from experiencescope.module.context_generator.base_context_generator import BaseContextGenerator
|
||||
from experiencescope.module.summarizer.base_summarizer import BaseSummarizer
|
||||
from experiencescope.schema.request import AgentWrapperRequest, ContextGeneratorRequest, SummarizerRequest
|
||||
from experiencescope.schema.response import AgentWrapperResponse, ContextGeneratorResponse, SummarizerResponse
|
||||
from experiencescope.schema.trajectory import ContextMessage, Trajectory, Sample
|
||||
from experiencemaker.module.agent_wrapper.base_agent_wrapper import BaseAgentWrapper
|
||||
from experiencemaker.module.context_generator.base_context_generator import BaseContextGenerator
|
||||
from experiencemaker.module.summarizer.base_summarizer import BaseSummarizer
|
||||
from experiencemaker.schema.request import AgentWrapperRequest, ContextGeneratorRequest, SummarizerRequest
|
||||
from experiencemaker.schema.response import AgentWrapperResponse, ContextGeneratorResponse, SummarizerResponse
|
||||
from experiencemaker.schema.trajectory import ContextMessage, Trajectory, Sample
|
||||
import uvicorn
|
||||
|
||||
app = FastAPI()
|
||||
|
|
@ -39,4 +39,4 @@ if __name__ == '__main__':
|
|||
uvicorn.run(app, host="0.0.0.0", port=8000, timeout_keep_alive=600000, limit_concurrency=32)
|
||||
|
||||
# launch with:
|
||||
# python -m experiencescope.service.model_service
|
||||
# python -m experiencemaker.service.model_service
|
||||
|
|
@ -1,8 +1,8 @@
|
|||
from pydantic import Field
|
||||
|
||||
from experiencescope.schema.request import AgentWrapperRequest, ContextGeneratorRequest, SummarizerRequest
|
||||
from experiencescope.schema.response import AgentWrapperResponse, ContextGeneratorResponse, SummarizerResponse
|
||||
from experiencescope.utils.http_client import HttpClient
|
||||
from experiencemaker.schema.request import AgentWrapperRequest, ContextGeneratorRequest, SummarizerRequest
|
||||
from experiencemaker.schema.response import AgentWrapperResponse, ContextGeneratorResponse, SummarizerResponse
|
||||
from experiencemaker.utils.http_client import HttpClient
|
||||
|
||||
|
||||
class ModelServiceClient(HttpClient):
|
||||
|
|
@ -2,7 +2,7 @@ from typing import List
|
|||
|
||||
from pydantic import BaseModel
|
||||
|
||||
from experiencescope.schema.trajectory import Sample
|
||||
from experiencemaker.schema.trajectory import Sample
|
||||
|
||||
|
||||
class BaseSampleBuffer(BaseModel):
|
||||
|
|
@ -3,8 +3,8 @@ from typing import List
|
|||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from experiencescope.model.base_embedding_model import BaseEmbeddingModel
|
||||
from experiencescope.schema.vector_store_node import VectorStoreNode
|
||||
from experiencemaker.model.base_embedding_model import BaseEmbeddingModel
|
||||
from experiencemaker.schema.vector_store_node import VectorStoreNode
|
||||
|
||||
|
||||
class BaseVectorStore(BaseModel, ABC):
|
||||
|
|
@ -6,8 +6,8 @@ from elasticsearch.helpers import bulk
|
|||
from loguru import logger
|
||||
from pydantic import Field, PrivateAttr, model_validator
|
||||
|
||||
from experiencescope.schema.vector_store_node import VectorStoreNode
|
||||
from experiencescope.storage.base_vector_store import BaseVectorStore
|
||||
from experiencemaker.schema.vector_store_node import VectorStoreNode
|
||||
from experiencemaker.storage.base_vector_store import BaseVectorStore
|
||||
|
||||
|
||||
class EsVectorStore(BaseVectorStore):
|
||||
|
|
@ -7,8 +7,8 @@ from typing import List, Any
|
|||
from loguru import logger
|
||||
from pydantic import Field, model_validator, PrivateAttr
|
||||
|
||||
from experiencescope.schema.vector_store_node import VectorStoreNode
|
||||
from experiencescope.storage.base_vector_store import BaseVectorStore
|
||||
from experiencemaker.schema.vector_store_node import VectorStoreNode
|
||||
from experiencemaker.storage.base_vector_store import BaseVectorStore
|
||||
|
||||
|
||||
class FileVectorStore(BaseVectorStore):
|
||||
10
experiencemaker/tool/__init__.py
Normal file
10
experiencemaker/tool/__init__.py
Normal file
|
|
@ -0,0 +1,10 @@
|
|||
from experiencemaker.tool.python_tools.code_tool import CodeTool
|
||||
from experiencemaker.tool.python_tools.dashscope_search_tool import DashscopeSearchTool
|
||||
from experiencemaker.tool.python_tools.terminate_tool import TerminateTool
|
||||
|
||||
from experiencemaker.utils.registry import Registry
|
||||
|
||||
TOOL_REGISTRY = Registry("tools")
|
||||
TOOL_REGISTRY.register(CodeTool)
|
||||
TOOL_REGISTRY.register(DashscopeSearchTool)
|
||||
TOOL_REGISTRY.register(TerminateTool)
|
||||
|
|
@ -1,7 +1,7 @@
|
|||
import sys
|
||||
from io import StringIO
|
||||
|
||||
from experiencescope.tool.base_tool import BaseTool
|
||||
from experiencemaker.tool.base_tool import BaseTool
|
||||
|
||||
|
||||
class CodeTool(BaseTool):
|
||||
|
|
@ -6,7 +6,7 @@ from dashscope.api_entities.dashscope_response import Message
|
|||
from loguru import logger
|
||||
from pydantic import Field
|
||||
|
||||
from experiencescope.tool.base_tool import BaseTool
|
||||
from experiencemaker.tool.base_tool import BaseTool
|
||||
|
||||
|
||||
class DashscopeSearchTool(BaseTool):
|
||||
|
|
@ -141,7 +141,7 @@ Extract the original content related to the user's question directly from the co
|
|||
return result
|
||||
|
||||
def main():
|
||||
from experiencescope.utils.test_key import set_key
|
||||
from experiencemaker.utils.test_key import set_key
|
||||
set_key()
|
||||
query = "What is artificial intelligence?"
|
||||
|
||||
|
|
@ -7,7 +7,7 @@ from mcp import ClientSession
|
|||
from mcp.client.sse import sse_client
|
||||
from pydantic import Field
|
||||
|
||||
from experiencescope.tool.base_tool import BaseTool
|
||||
from experiencemaker.tool.base_tool import BaseTool
|
||||
|
||||
|
||||
class MCPTool(BaseTool):
|
||||
|
|
@ -1,4 +1,4 @@
|
|||
from experiencescope.tool.base_tool import BaseTool
|
||||
from experiencemaker.tool.base_tool import BaseTool
|
||||
|
||||
|
||||
class TerminateTool(BaseTool):
|
||||
|
|
@ -6,7 +6,7 @@ import requests
|
|||
from loguru import logger
|
||||
from pydantic import BaseModel, Field, PrivateAttr
|
||||
|
||||
from experiencescope.enumeration.http_enum import HttpEnum
|
||||
from experiencemaker.enumeration.http_enum import HttpEnum
|
||||
|
||||
|
||||
class HttpClient(BaseModel):
|
||||
|
|
@ -1,7 +1,7 @@
|
|||
import json
|
||||
import os
|
||||
|
||||
from experiencescope.schema.module_loader import ModuleLoader
|
||||
from experiencemaker.schema.module_loader import ModuleLoader
|
||||
|
||||
|
||||
def load_env_keys():
|
||||
|
|
@ -13,21 +13,21 @@ def load_env_keys():
|
|||
|
||||
|
||||
agent_wrapper_loader = ModuleLoader(
|
||||
class_path="experiencescope.module.agent_wrapper.naive_agent_wrapper",
|
||||
class_path="experiencemaker.module.agent_wrapper.naive_agent_wrapper",
|
||||
class_name="NaiveAgentWrapper",
|
||||
config_path="beyondagent/config/agent_wrapper/naive_agent_wrapper.json")
|
||||
|
||||
context_generator_loader = ModuleLoader(
|
||||
class_path="experiencescope.module.context_generator.simple_context_generator",
|
||||
class_path="experiencemaker.module.context_generator.simple_context_generator",
|
||||
class_name="SimpleContextGenerator",
|
||||
config_path="beyondagent/config/context_generator/simple_context_generator.json")
|
||||
|
||||
summarizer_loader = ModuleLoader(
|
||||
class_path="experiencescope.module.summarizer.simple_summarizer",
|
||||
class_path="experiencemaker.module.summarizer.simple_summarizer",
|
||||
class_name="SimpleSummarizer",
|
||||
config_path="beyondagent/config/summarizer/simple_summarizer.json")
|
||||
|
||||
env_loader = ModuleLoader(
|
||||
class_path="experiencescope.module.environment.simple_environment",
|
||||
class_path="experiencemaker.module.environment.simple_environment",
|
||||
class_name="SimpleEnvironment",
|
||||
config_path="beyondagent/config/environment/simple_environment.json")
|
||||
|
|
@ -1,6 +1,6 @@
|
|||
from typing import List
|
||||
|
||||
from experiencescope.schema.trajectory import Message, StateMessage, ActionMessage
|
||||
from experiencemaker.schema.trajectory import Message, StateMessage, ActionMessage
|
||||
|
||||
|
||||
def format_trajectory_steps(steps: List[Message]) -> str:
|
||||
|
|
@ -1,10 +0,0 @@
|
|||
from experiencescope.tool.python_tools.code_tool import CodeTool
|
||||
from experiencescope.tool.python_tools.dashscope_search_tool import DashscopeSearchTool
|
||||
from experiencescope.tool.python_tools.terminate_tool import TerminateTool
|
||||
|
||||
from experiencescope.utils.registry import Registry
|
||||
|
||||
TOOL_REGISTRY = Registry("tools")
|
||||
TOOL_REGISTRY.register(CodeTool)
|
||||
TOOL_REGISTRY.register(DashscopeSearchTool)
|
||||
TOOL_REGISTRY.register(TerminateTool)
|
||||
Loading…
Add table
Reference in a new issue