Skip to content

Commit c851b63

Browse files
committed
Add flake8
1 parent f06f191 commit c851b63

File tree

7 files changed

+56
-20
lines changed

7 files changed

+56
-20
lines changed

.flake8

+36
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
[flake8]
2+
# Maximum line length
3+
max-line-length = 160
4+
# Maximum complexity
5+
max-complexity = 10
6+
# Files to exclude
7+
exclude =
8+
.git,
9+
__pycache__,
10+
build,
11+
dist,
12+
*.egg-info,
13+
venv
14+
15+
# Ignore specific errors
16+
ignore =
17+
# Missing docstring in public module
18+
D100,
19+
# Missing docstring in public class
20+
D101,
21+
# Missing docstring in public method
22+
D102,
23+
# Missing docstring in public function
24+
D103,
25+
# Line break before binary operator
26+
W503,
27+
# Line break after binary operator
28+
W504
29+
30+
# Specify the order of imports
31+
import-order-style = google
32+
33+
# Additional configurations
34+
statistics = True
35+
count = True
36+
show-source = True

.github/workflows/pylint.yml .github/workflows/flake8.yml

+4-6
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
name: Pylint
1+
name: Flake8
22

3-
on: [push]
3+
on: [push, pull_request]
44

55
jobs:
66
build:
@@ -19,10 +19,8 @@ jobs:
1919
- name: Install dependencies
2020
run: |
2121
python -m pip install --upgrade pip
22-
pip install -r src/requirements.txt
23-
pip install pylint
22+
pip install flake8 flake8-import-order flake8-bugbear
2423
2524
- name: Analysing the code with pylint
26-
working-directory: src
2725
run: |
28-
pylint $(git ls-files '*.py')
26+
flake8 src/

.vscode/settings.json

+5
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
{
2+
"flake8.args": [
3+
"--config=.flake8"
4+
]
5+
}

src/agents/tool_use_agent.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,17 @@
11
from typing import List
22

33
from autogen_core.base import AgentId, MessageContext
4-
from autogen_core.components import RoutedAgent, message_handler
4+
from autogen_core.components import message_handler, RoutedAgent
5+
from autogen_core.components.model_context import BufferedChatCompletionContext
56
from autogen_core.components.models import (
7+
AssistantMessage,
68
ChatCompletionClient,
79
LLMMessage,
810
SystemMessage,
911
UserMessage,
1012
)
1113
from autogen_core.components.tool_agent import tool_agent_caller_loop
1214
from autogen_core.components.tools import ToolSchema
13-
from autogen_core.components.model_context import BufferedChatCompletionContext
14-
from autogen_core.components.models import AssistantMessage
15-
16-
1715
from src.message_protocol.messages import Message
1816

1917

src/main.py

+5-7
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,17 @@
11
import asyncio
22

3-
from .agents.tool_use_agent import ToolUseAgent
4-
5-
from .tools.tool_factory import get_tools
63
from autogen_core.application import SingleThreadedAgentRuntime
4+
from autogen_core.base import AgentId
75
from autogen_core.components.tool_agent import ToolAgent
86
from autogen_ext.models import OpenAIChatCompletionClient
9-
from autogen_core.base import AgentId
10-
from .message_protocol.messages import Message
11-
127
from dotenv import load_dotenv
13-
148
from rich.console import Console
159
from rich.markdown import Markdown
1610

11+
from .agents.tool_use_agent import ToolUseAgent
12+
from .message_protocol.messages import Message
13+
from .tools.tool_factory import get_tools
14+
1715

1816
async def main():
1917
load_dotenv()

src/message_protocol/messages.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from dataclasses import dataclass
22

3+
34
@dataclass
45
class Message:
5-
content: str
6+
content: str

src/tools/tool_factory.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from langchain_google_community import GmailToolkit
21
from autogen_ext.tools import LangChainToolAdapter
2+
from langchain_google_community import GmailToolkit
33

44

55
def get_tools():

0 commit comments

Comments
 (0)