apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-07-23 17:01:08 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update config int

Browse source
This commit is contained in:
张建生 2025-02-28 16:36:29 +08:00
parent c4f169462f
commit 632e14d415
20 changed files with 21 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

3
metagpt/rag/engines/simple.py
View file

@ -31,7 +31,7 @@ from llama_index.core.schema import (
TransformComponent,
)
from metagpt.config2 import Config
from metagpt.config2 import config
from metagpt.rag.factories import (
get_index,
get_rag_embedding,
@ -400,7 +400,6 @@ class SimpleEngine(RetrieverQueryEngine):
dict[file_type: BaseReader]
"""
file_extractor: dict[str:BaseReader] = {}
config = Config.default()
if config.omniparse.base_url:
pdf_parser = OmniParse(
api_key=config.omniparse.api_key,

3
metagpt/rag/factories/llm.py
View file

@ -12,7 +12,7 @@ from llama_index.core.llms import (
from llama_index.core.llms.callbacks import llm_completion_callback
from pydantic import Field
from metagpt.config2 import Config
from metagpt.config2 import config
from metagpt.provider.base_llm import BaseLLM
from metagpt.utils.async_helper import NestAsyncio
from metagpt.utils.token_counter import TOKEN_MAX
@ -41,7 +41,6 @@ class RAGLLM(CustomLLM):
**kwargs
):
super().__init__(*args, **kwargs)
config = Config.default()
if context_window < 0:
context_window = TOKEN_MAX.get(config.llm.model, DEFAULT_CONTEXT_WINDOW)

4
metagpt/rag/schema.py
View file

@ -11,7 +11,7 @@ from llama_index.core.schema import TextNode
from llama_index.core.vector_stores.types import VectorStoreQueryMode
from pydantic import BaseModel, ConfigDict, Field, PrivateAttr, model_validator
from metagpt.config2 import Config
from metagpt.config2 import config
from metagpt.configs.embedding_config import EmbeddingType
from metagpt.logs import logger
from metagpt.rag.interface import RAGObject
@ -47,7 +47,6 @@ class FAISSRetrieverConfig(IndexRetrieverConfig):
@model_validator(mode="after")
def check_dimensions(self):
if self.dimensions == 0:
config = Config.default()
self.dimensions = config.embedding.dimensions or self._embedding_type_to_dimensions.get(
config.embedding.api_type, 1536
)
@ -89,7 +88,6 @@ class MilvusRetrieverConfig(IndexRetrieverConfig):
@model_validator(mode="after")
def check_dimensions(self):
if self.dimensions == 0:
config = Config.default()
self.dimensions = config.embedding.dimensions or self._embedding_type_to_dimensions.get(
config.embedding.api_type, 1536
)