From fe2ca9d5f5f19a060e3f9ac0671d8d030445f334 Mon Sep 17 00:00:00 2001 From: Siddhesh Agarwal Date: Sun, 10 Nov 2024 20:00:13 +0530 Subject: [PATCH] black + isort formatting --- examples/cooking_recipe_example.py | 2 +- examples/discussion.py | 3 +- examples/enhanced_context.py | 47 +++++++++++++----------------- examples/four_way_conversation.py | 1 + examples/inspiration_plugin.py | 1 + examples/medicine_data.py | 2 +- examples/mood_detector_plugin.py | 2 +- examples/sentiment_analysis.py | 1 + examples/web_bible_explorer.py | 10 ++++--- simplemind/settings.py | 3 +- tests/test_conversations.py | 3 +- tests/test_generate_data.py | 5 ++-- tests/test_generate_text.py | 2 +- 13 files changed, 39 insertions(+), 43 deletions(-) diff --git a/examples/cooking_recipe_example.py b/examples/cooking_recipe_example.py index 8abc295..077d9ae 100644 --- a/examples/cooking_recipe_example.py +++ b/examples/cooking_recipe_example.py @@ -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 diff --git a/examples/discussion.py b/examples/discussion.py index d63353d..42781b3 100644 --- a/examples/discussion.py +++ b/examples/discussion.py @@ -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.""" diff --git a/examples/enhanced_context.py b/examples/enhanced_context.py index a40fd1f..ba63f8a 100644 --- a/examples/enhanced_context.py +++ b/examples/enhanced_context.py @@ -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"] diff --git a/examples/four_way_conversation.py b/examples/four_way_conversation.py index c414e46..2ee283a 100644 --- a/examples/four_way_conversation.py +++ b/examples/four_way_conversation.py @@ -1,4 +1,5 @@ import time + from _context import simplemind as sm diff --git a/examples/inspiration_plugin.py b/examples/inspiration_plugin.py index db2001b..6180385 100644 --- a/examples/inspiration_plugin.py +++ b/examples/inspiration_plugin.py @@ -1,4 +1,5 @@ import random + from _context import simplemind as sm diff --git a/examples/medicine_data.py b/examples/medicine_data.py index b6a7be4..361e833 100644 --- a/examples/medicine_data.py +++ b/examples/medicine_data.py @@ -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 diff --git a/examples/mood_detector_plugin.py b/examples/mood_detector_plugin.py index 5f61c36..e15c21e 100644 --- a/examples/mood_detector_plugin.py +++ b/examples/mood_detector_plugin.py @@ -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") diff --git a/examples/sentiment_analysis.py b/examples/sentiment_analysis.py index 8354046..2c380fa 100644 --- a/examples/sentiment_analysis.py +++ b/examples/sentiment_analysis.py @@ -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 diff --git a/examples/web_bible_explorer.py b/examples/web_bible_explorer.py index 4012078..373c650 100644 --- a/examples/web_bible_explorer.py +++ b/examples/web_bible_explorer.py @@ -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() diff --git a/simplemind/settings.py b/simplemind/settings.py index ee11063..de69339 100644 --- a/simplemind/settings.py +++ b/simplemind/settings.py @@ -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`" diff --git a/tests/test_conversations.py b/tests/test_conversations.py index b0c85dd..4747f65 100644 --- a/tests/test_conversations.py +++ b/tests/test_conversations.py @@ -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( diff --git a/tests/test_generate_data.py b/tests/test_generate_data.py index 610c96a..cd00bc6 100644 --- a/tests/test_generate_data.py +++ b/tests/test_generate_data.py @@ -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 diff --git a/tests/test_generate_text.py b/tests/test_generate_text.py index 4ab62cf..2b3eb0e 100644 --- a/tests/test_generate_text.py +++ b/tests/test_generate_text.py @@ -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(