Megumin
2 months ago
committed by
GitHub
parent
cdc720df8c
commit
6114f39861
@ -0,0 +1,135 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2025 CloudWeGo Authors
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/cloudwego/eino/components/tool"
|
||||||
|
"github.com/cloudwego/eino/components/tool/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
type GetWeatherInput struct {
|
||||||
|
Location string `json:"location" jsonschema:"description=The city and state, e.g. San Francisco, CA"`
|
||||||
|
Unit string `json:"unit,omitempty" jsonschema:"enum=celsius,enum=fahrenheit,description=The unit of temperature"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetWeatherOutput struct {
|
||||||
|
Temperature float64 `json:"temperature"`
|
||||||
|
Unit string `json:"unit"`
|
||||||
|
Condition string `json:"condition"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetForecastInput struct {
|
||||||
|
Location string `json:"location" jsonschema:"description=The city and state"`
|
||||||
|
Days int `json:"days" jsonschema:"description=Number of days to forecast (1-10)"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetForecastOutput struct {
|
||||||
|
Forecasts []DayForecast `json:"forecasts"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type DayForecast struct {
|
||||||
|
Day string `json:"day"`
|
||||||
|
Temperature float64 `json:"temperature"`
|
||||||
|
Condition string `json:"condition"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetStockPriceInput struct {
|
||||||
|
Ticker string `json:"ticker" jsonschema:"description=Stock ticker symbol (e.g., AAPL, GOOGL)"`
|
||||||
|
IncludeHistory bool `json:"include_history,omitempty" jsonschema:"description=Include historical data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetStockPriceOutput struct {
|
||||||
|
Ticker string `json:"ticker"`
|
||||||
|
Price float64 `json:"price"`
|
||||||
|
Change float64 `json:"change"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ConvertCurrencyInput struct {
|
||||||
|
Amount float64 `json:"amount" jsonschema:"description=Amount to convert"`
|
||||||
|
FromCurrency string `json:"from_currency" jsonschema:"description=Source currency code (e.g., USD)"`
|
||||||
|
ToCurrency string `json:"to_currency" jsonschema:"description=Target currency code (e.g., EUR)"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ConvertCurrencyOutput struct {
|
||||||
|
OriginalAmount float64 `json:"original_amount"`
|
||||||
|
ConvertedAmount float64 `json:"converted_amount"`
|
||||||
|
ExchangeRate float64 `json:"exchange_rate"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func createWeatherTools() []tool.BaseTool {
|
||||||
|
getWeather, _ := utils.InferTool(
|
||||||
|
"get_weather",
|
||||||
|
"Get the current weather in a given location",
|
||||||
|
func(ctx context.Context, input *GetWeatherInput) (*GetWeatherOutput, error) {
|
||||||
|
return &GetWeatherOutput{
|
||||||
|
Temperature: 22.5,
|
||||||
|
Unit: input.Unit,
|
||||||
|
Condition: "Sunny",
|
||||||
|
}, nil
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
getForecast, _ := utils.InferTool(
|
||||||
|
"get_forecast",
|
||||||
|
"Get the weather forecast for multiple days ahead",
|
||||||
|
func(ctx context.Context, input *GetForecastInput) (*GetForecastOutput, error) {
|
||||||
|
forecasts := make([]DayForecast, input.Days)
|
||||||
|
for i := 0; i < input.Days; i++ {
|
||||||
|
forecasts[i] = DayForecast{
|
||||||
|
Day: fmt.Sprintf("Day %d", i+1),
|
||||||
|
Temperature: 20.0 + float64(i),
|
||||||
|
Condition: "Partly Cloudy",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return &GetForecastOutput{Forecasts: forecasts}, nil
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
return []tool.BaseTool{getWeather, getForecast}
|
||||||
|
}
|
||||||
|
|
||||||
|
func createFinanceTools() []tool.BaseTool {
|
||||||
|
getStockPrice, _ := utils.InferTool(
|
||||||
|
"get_stock_price",
|
||||||
|
"Get the current stock price and market data for a given ticker symbol",
|
||||||
|
func(ctx context.Context, input *GetStockPriceInput) (*GetStockPriceOutput, error) {
|
||||||
|
return &GetStockPriceOutput{
|
||||||
|
Ticker: input.Ticker,
|
||||||
|
Price: 150.25,
|
||||||
|
Change: 2.5,
|
||||||
|
}, nil
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
convertCurrency, _ := utils.InferTool(
|
||||||
|
"convert_currency",
|
||||||
|
"Convert an amount from one currency to another using current exchange rates",
|
||||||
|
func(ctx context.Context, input *ConvertCurrencyInput) (*ConvertCurrencyOutput, error) {
|
||||||
|
rate := 0.85
|
||||||
|
return &ConvertCurrencyOutput{
|
||||||
|
OriginalAmount: input.Amount,
|
||||||
|
ConvertedAmount: input.Amount * rate,
|
||||||
|
ExchangeRate: rate,
|
||||||
|
}, nil
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
return []tool.BaseTool{getStockPrice, convertCurrency}
|
||||||
|
}
|
||||||
@ -0,0 +1,81 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2025 CloudWeGo Authors
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/cloudwego/eino-ext/components/model/ark"
|
||||||
|
"github.com/cloudwego/eino/adk"
|
||||||
|
"github.com/cloudwego/eino/adk/middlewares/dynamictool/toolsearch"
|
||||||
|
|
||||||
|
"github.com/cloudwego/eino-examples/adk/common/prints"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
weatherTools := createWeatherTools()
|
||||||
|
financeTools := createFinanceTools()
|
||||||
|
allDynamicTools := append(weatherTools, financeTools...)
|
||||||
|
|
||||||
|
toolSearchMiddleware, err := toolsearch.New(ctx, &toolsearch.Config{
|
||||||
|
DynamicTools: allDynamicTools,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("failed to create tool search middleware: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
chatModel, err := ark.NewChatModel(ctx, &ark.ChatModelConfig{
|
||||||
|
APIKey: os.Getenv("ARK_API_KEY"),
|
||||||
|
Model: os.Getenv("ARK_MODEL"),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
agent, err := adk.NewChatModelAgent(ctx, &adk.ChatModelAgentConfig{
|
||||||
|
Name: "tool_search_agent",
|
||||||
|
Description: "An agent that can dynamically search and use tools from a large tool library",
|
||||||
|
Instruction: `You are a helpful assistant.`,
|
||||||
|
Model: chatModel,
|
||||||
|
Handlers: []adk.ChatModelAgentMiddleware{
|
||||||
|
toolSearchMiddleware,
|
||||||
|
},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("failed to create agent: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
runner := adk.NewRunner(ctx, adk.RunnerConfig{
|
||||||
|
Agent: agent,
|
||||||
|
EnableStreaming: true,
|
||||||
|
})
|
||||||
|
iter := runner.Query(ctx, "What's the weather in Beijing?")
|
||||||
|
for {
|
||||||
|
event, ok := iter.Next()
|
||||||
|
if !ok {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
prints.Event(event)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,105 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2025 CloudWeGo Authors
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
|
"github.com/cloudwego/eino-ext/adk/backend/local"
|
||||||
|
"github.com/cloudwego/eino-ext/components/model/ark"
|
||||||
|
"github.com/cloudwego/eino/adk"
|
||||||
|
"github.com/cloudwego/eino/adk/middlewares/filesystem"
|
||||||
|
"github.com/cloudwego/eino/adk/middlewares/skill"
|
||||||
|
|
||||||
|
"github.com/cloudwego/eino-examples/adk/common/prints"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
ctx := context.Background()
|
||||||
|
pwd, _ := os.Getwd()
|
||||||
|
workDir := filepath.Join(pwd, "adk", "middlewares", "skill", "workdir")
|
||||||
|
skillsDir := filepath.Join(workDir, "skills")
|
||||||
|
|
||||||
|
cm, err := ark.NewChatModel(ctx, &ark.ChatModelConfig{
|
||||||
|
APIKey: os.Getenv("ARK_API_KEY"),
|
||||||
|
Model: os.Getenv("ARK_MODEL"),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
be, err := local.NewBackend(ctx, &local.Config{})
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
fsm, err := filesystem.New(ctx, &filesystem.MiddlewareConfig{
|
||||||
|
Backend: be,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
skillBackend, err := skill.NewBackendFromFilesystem(ctx, &skill.BackendFromFilesystemConfig{
|
||||||
|
Backend: be,
|
||||||
|
BaseDir: skillsDir,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Failed to create skill backend: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
sm, err := skill.NewMiddleware(ctx, &skill.Config{
|
||||||
|
Backend: skillBackend,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Failed to create skill middleware: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
agent, err := adk.NewChatModelAgent(ctx, &adk.ChatModelAgentConfig{
|
||||||
|
Name: "LogAnalysisAgent",
|
||||||
|
Description: "An agent that can analyze logs",
|
||||||
|
Instruction: "You are a helpful assistant.",
|
||||||
|
Model: cm,
|
||||||
|
Handlers: []adk.ChatModelAgentMiddleware{fsm, sm},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Failed to create agent: %v", err)
|
||||||
|
}
|
||||||
|
runner := adk.NewRunner(ctx, adk.RunnerConfig{
|
||||||
|
Agent: agent,
|
||||||
|
})
|
||||||
|
|
||||||
|
input := fmt.Sprintf("Analyze the %s file", filepath.Join(workDir, "test.log"))
|
||||||
|
log.Println("User: ", input)
|
||||||
|
|
||||||
|
iterator := runner.Query(ctx, input)
|
||||||
|
for {
|
||||||
|
event, ok := iterator.Next()
|
||||||
|
if !ok {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if event.Err != nil {
|
||||||
|
log.Printf("Error: %v\n", event.Err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
prints.Event(event)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,24 @@
|
|||||||
|
---
|
||||||
|
name: log_analyzer
|
||||||
|
description: A skill to analyze log files for errors and warnings using a Python script.
|
||||||
|
---
|
||||||
|
|
||||||
|
# Log Analyzer Skill
|
||||||
|
|
||||||
|
This skill analyzes a log file to count the occurrences of "ERROR" and "WARNING" and lists the lines where they appear.
|
||||||
|
|
||||||
|
## Capability
|
||||||
|
|
||||||
|
The skill provides a Python script named `analyze.py` located in this directory. You can use this script to analyze any text file.
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
To use this skill, execute the `analyze.py` script with the target log file as an argument.
|
||||||
|
|
||||||
|
### Example
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python3 {{.BaseDirectory}}/scripts/analyze.py /path/to/logfile.log
|
||||||
|
```
|
||||||
|
|
||||||
|
**Note**: Replace `/path/to/logfile.log` with the actual path of the file you want to analyze.
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
import sys
|
||||||
|
import os
|
||||||
|
|
||||||
|
def analyze_log(file_path):
|
||||||
|
if not os.path.exists(file_path):
|
||||||
|
print(f"Error: File '{file_path}' not found.")
|
||||||
|
return
|
||||||
|
|
||||||
|
error_count = 0
|
||||||
|
warning_count = 0
|
||||||
|
error_lines = []
|
||||||
|
warning_lines = []
|
||||||
|
|
||||||
|
try:
|
||||||
|
with open(file_path, 'r') as f:
|
||||||
|
for i, line in enumerate(f, 1):
|
||||||
|
content = line.strip()
|
||||||
|
if "ERROR" in content:
|
||||||
|
error_count += 1
|
||||||
|
error_lines.append(f"Line {i}: {content}")
|
||||||
|
elif "WARNING" in content:
|
||||||
|
warning_count += 1
|
||||||
|
warning_lines.append(f"Line {i}: {content}")
|
||||||
|
|
||||||
|
print(f"Analysis Result for {file_path}:")
|
||||||
|
print(f"Total Errors: {error_count}")
|
||||||
|
print(f"Total Warnings: {warning_count}")
|
||||||
|
|
||||||
|
if error_count > 0:
|
||||||
|
print("\nError Details:")
|
||||||
|
for line in error_lines:
|
||||||
|
print(line)
|
||||||
|
|
||||||
|
if warning_count > 0:
|
||||||
|
print("\nWarning Details:")
|
||||||
|
for line in warning_lines:
|
||||||
|
print(line)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f"An error occurred while reading the file: {e}")
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
if len(sys.argv) < 2:
|
||||||
|
print("Usage: python3 analyze.py <log_file_path>")
|
||||||
|
else:
|
||||||
|
analyze_log(sys.argv[1])
|
||||||
@ -0,0 +1,6 @@
|
|||||||
|
[2024-05-20 10:00:00] INFO: Service started.
|
||||||
|
[2024-05-20 10:01:23] WARNING: High memory usage detected.
|
||||||
|
[2024-05-20 10:02:15] ERROR: Database connection failed.
|
||||||
|
[2024-05-20 10:03:00] INFO: Retry connection.
|
||||||
|
[2024-05-20 10:03:05] ERROR: Connection timed out.
|
||||||
|
[2024-05-20 10:05:00] INFO: Service stopped.
|
||||||
Loading…
Reference in New Issue