Code Formatting (#1828)

This commit is contained in:
Dev Khant
2024-09-07 22:39:28 +05:30
committed by GitHub
parent 6a54d27286
commit a972d2fb07
66 changed files with 208 additions and 138 deletions

View File

@@ -1,7 +1,4 @@
import subprocess
import sys
import os
import json
from typing import Dict, List, Optional
try:
@@ -9,8 +6,8 @@ try:
except ImportError:
raise ImportError("The 'anthropic' library is required. Please install it using 'pip install anthropic'.")
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class AnthropicLLM(LLMBase):

View File

@@ -1,16 +1,14 @@
import subprocess
import sys
import os
import json
from typing import Dict, List, Optional, Any
import os
from typing import Any, Dict, List, Optional
try:
import boto3
except ImportError:
raise ImportError("The 'boto3' library is required. Please install it using 'pip install boto3'.")
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class AWSBedrockLLM(LLMBase):

View File

@@ -1,11 +1,11 @@
import os
import json
import os
from typing import Dict, List, Optional
from openai import AzureOpenAI
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class AzureOpenAILLM(LLMBase):

View File

@@ -1,5 +1,5 @@
from typing import Optional
from abc import ABC, abstractmethod
from typing import Optional
from mem0.configs.llms.base import BaseLlmConfig

View File

@@ -1,7 +1,5 @@
import subprocess
import sys
import os
import json
import os
from typing import Dict, List, Optional
try:
@@ -9,8 +7,8 @@ try:
except ImportError:
raise ImportError("The 'groq' library is required. Please install it using 'pip install groq'.")
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class GroqLLM(LLMBase):

View File

@@ -1,5 +1,3 @@
import subprocess
import sys
import json
from typing import Dict, List, Optional
@@ -8,8 +6,8 @@ try:
except ImportError:
raise ImportError("The 'litellm' library is required. Please install it using 'pip install litellm'.")
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class LiteLLM(LLMBase):

View File

@@ -1,5 +1,3 @@
import subprocess
import sys
from typing import Dict, List, Optional
try:
@@ -7,8 +5,8 @@ try:
except ImportError:
raise ImportError("The 'ollama' library is required. Please install it using 'pip install ollama'.")
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class OllamaLLM(LLMBase):

View File

@@ -1,11 +1,11 @@
import os
import json
import os
from typing import Dict, List, Optional
from openai import OpenAI
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class OpenAILLM(LLMBase):

View File

@@ -1,10 +1,11 @@
import os, json
import json
import os
from typing import Dict, List, Optional
from openai import OpenAI
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class OpenAIStructuredLLM(LLMBase):

View File

@@ -1,7 +1,5 @@
import subprocess
import sys
import os
import json
import os
from typing import Dict, List, Optional
try:
@@ -9,8 +7,8 @@ try:
except ImportError:
raise ImportError("The 'together' library is required. Please install it using 'pip install together'.")
from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig
from mem0.llms.base import LLMBase
class TogetherLLM(LLMBase):