black + isort formatting

This commit is contained in:
Siddhesh Agarwal
2024-11-10 20:00:13 +05:30
parent 670240b943
commit fe2ca9d5f5
13 changed files with 39 additions and 43 deletions
+1 -1
View File
@@ -1,5 +1,5 @@
from pydantic import BaseModel
from _context import simplemind as sm
from pydantic import BaseModel
from rich.console import Console
from rich.panel import Panel
from rich.text import Text
+1 -2
View File
@@ -1,11 +1,10 @@
import time
from typing import List, Tuple
from _context import sm
from rich.console import Console
from rich.markdown import Markdown
from _context import sm
class MultiAIConversation:
"""Orchestrates conversations between multiple AI models."""
+20 -27
View File
@@ -1,35 +1,28 @@
from datetime import datetime
import logging
import sqlite3
from typing import List
import re
import os
import contextlib
import spacy
import logging
import os
import random
import re
import sqlite3
from concurrent.futures import ThreadPoolExecutor
from contextlib import contextmanager
from _context import simplemind as sm
from datetime import datetime
from typing import List
import nltk
from nltk.tokenize import word_tokenize
from nltk.tag import pos_tag
from rich.console import Console
from rich.panel import Panel
from rich.markdown import Markdown
from rich.status import Status
from concurrent.futures import ThreadPoolExecutor
import random
from docopt import docopt
from prompt_toolkit import PromptSession
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
import spacy
import xerox
from _context import simplemind as sm
from docopt import docopt
from nltk.tag import pos_tag
from nltk.tokenize import word_tokenize
from prompt_toolkit import PromptSession
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
from prompt_toolkit.completion import Completer, Completion
from rich.console import Console
from rich.markdown import Markdown
from rich.panel import Panel
from rich.status import Status
DB_PATH = "enhanced_context.db"
AVAILABLE_PROVIDERS = ["xai", "openai", "anthropic", "ollama"]
+1
View File
@@ -1,4 +1,5 @@
import time
from _context import simplemind as sm
+1
View File
@@ -1,4 +1,5 @@
import random
from _context import simplemind as sm
+1 -1
View File
@@ -1,5 +1,5 @@
from pydantic import BaseModel
from _context import simplemind as sm
from pydantic import BaseModel
from rich.console import Console
from rich.panel import Panel
from rich.table import Table
+1 -1
View File
@@ -1,7 +1,7 @@
import nltk
from _context import simplemind as sm
from nltk.sentiment import SentimentIntensityAnalyzer
from rich.console import Console
from _context import simplemind as sm
nltk.download("vader_lexicon")
+1
View File
@@ -5,6 +5,7 @@ from pydantic import BaseModel
# Note: you should probably be using textblob for this.
class SentimentAnalysis(BaseModel):
sentiment: Literal["positive", "negative", "neutral"]
confidence: float
+6 -4
View File
@@ -1,8 +1,10 @@
from fastapi import FastAPI, Request, HTTPException
from fastapi.templating import Jinja2Templates
from fastapi.staticfiles import StaticFiles
from pydantic import BaseModel
from typing import List
from fastapi import FastAPI, HTTPException, Request
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from pydantic import BaseModel
import simplemind as sm
app = FastAPI()
+2 -1
View File
@@ -14,8 +14,9 @@ class LoggingConfig(BaseSettings):
"""Enable logging for the application."""
# adding imports here to avoid forced dependencies
try:
import logfire
from logging import basicConfig
import logfire
except ImportError as e:
raise ImportError(
"To enable logging, please install logfire: `pip install logfire`"
+1 -2
View File
@@ -1,8 +1,7 @@
import pytest
from simplemind.providers import Anthropic, Gemini, OpenAI, Groq, Ollama, Amazon
import simplemind as sm
from simplemind.providers import Amazon, Anthropic, Gemini, Groq, Ollama, OpenAI
@pytest.mark.parametrize(
+2 -3
View File
@@ -1,9 +1,8 @@
import pytest
from simplemind.providers import Anthropic, Gemini, OpenAI, Groq, Ollama, Amazon
from pydantic import BaseModel
from simplemind.providers import Amazon, Anthropic, Gemini, Groq, Ollama, OpenAI
class ResponseModel(BaseModel):
result: int
+1 -1
View File
@@ -1,6 +1,6 @@
import pytest
from simplemind.providers import Anthropic, Gemini, OpenAI, Groq, Ollama, Amazon
from simplemind.providers import Amazon, Anthropic, Gemini, Groq, Ollama, OpenAI
@pytest.mark.parametrize(