Forge: Moved memory and llm out of the sdk

This commit is contained in:
SwiftyOS
2023-12-06 15:34:08 +01:00
parent ccb700d00a
commit 2c4184b81e
7 changed files with 3 additions and 9 deletions

View File

@@ -7,13 +7,9 @@ from forge.sdk import (
Task,
TaskRequestBody,
Workspace,
PromptEngine,
chat_completion_request,
ChromaMemStore,
)
from forge.actions import ActionRegister
import json
import pprint
LOG = ForgeLogger(__name__)

View File

@@ -3,7 +3,7 @@ import typing
import openai
from tenacity import retry, stop_after_attempt, wait_random_exponential
from .forge_log import ForgeLogger
from .sdk.forge_log import ForgeLogger
from litellm import completion, acompletion, AuthenticationError, InvalidRequestError
LOG = ForgeLogger(__name__)

View File

@@ -5,7 +5,7 @@ core of the Forge.
from .agent import Agent
from .db import AgentDB, Base
from .forge_log import ForgeLogger
from .llm import chat_completion_request, create_embedding_request, transcribe_audio
from ..llm import chat_completion_request, create_embedding_request, transcribe_audio
from .prompting import PromptEngine
from .model import (
Artifact,
@@ -23,5 +23,3 @@ from .model import (
)
from .workspace import LocalWorkspace, Workspace
from .errors import *
from .memory.chroma_memstore import ChromaMemStore
from .memory.memstore import MemStore