Skip to content

Commit

Permalink
fix: removed spammy warning
Browse files Browse the repository at this point in the history
  • Loading branch information
ErikBjare committed Aug 20, 2024
1 parent c43cc8a commit 90aa4a6
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions gptme/llm.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@

from rich import print

from .config import get_config
from .constants import PROMPT_ASSISTANT
from .llm_anthropic import chat as chat_anthropic
from .llm_anthropic import get_client as get_anthropic_client
from .llm_anthropic import init as init_anthropic
Expand All @@ -14,8 +16,6 @@
from .llm_openai import get_client as get_openai_client
from .llm_openai import init as init_openai
from .llm_openai import stream as stream_openai
from .config import get_config
from .constants import PROMPT_ASSISTANT
from .message import Message, len_tokens
from .models import MODELS, get_summary_model
from .util import extract_codeblocks
Expand Down Expand Up @@ -101,10 +101,6 @@ def print_clear():
if is_supported_codeblock_tool(lang):
print("\nFound codeblock, breaking")
break
else:
logger.warning(
"Code block not supported by tools, continuing generation"
)
except KeyboardInterrupt:
return Message("assistant", output + "... ^C Interrupted")
finally:
Expand Down

0 comments on commit 90aa4a6

Please sign in to comment.