Skip to content

Commit f6aa076

Browse files
committed
updates to examples, fix on streaming tools
Signed-off-by: Filinto Duran <[email protected]>
1 parent e37ca41 commit f6aa076

File tree

4 files changed

+22
-35
lines changed

4 files changed

+22
-35
lines changed

dapr/aio/clients/grpc/client.py

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -2015,20 +2015,7 @@ def convert_content_part_to_pb(part):
20152015
elif response.HasField('complete'):
20162016
# Handle completion
20172017
complete_pb = response.complete
2018-
2019-
usage = None
2020-
if complete_pb.HasField('usage'):
2021-
usage = ConversationUsage(
2022-
prompt_tokens=complete_pb.usage.prompt_tokens,
2023-
completion_tokens=complete_pb.usage.completion_tokens,
2024-
total_tokens=complete_pb.usage.total_tokens
2025-
)
2026-
2027-
complete = ConversationStreamComplete(
2028-
context_id=complete_pb.contextID if complete_pb.HasField('contextID') else None,
2029-
usage=usage
2030-
)
2031-
2018+
complete = ConversationStreamComplete.from_proto(complete_pb)
20322019
yield ConversationStreamResponse(complete=complete)
20332020

20342021
except grpc.aio.AioRpcError as err:

dapr/clients/grpc/client.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
21
"""
32
Copyright 2023 The Dapr Authors
43
Licensed under the Apache License, Version 2.0 (the "License");
@@ -2050,20 +2049,7 @@ def convert_content_part_to_pb(part):
20502049
elif response.HasField('complete'):
20512050
# Handle completion
20522051
complete_pb = response.complete
2053-
2054-
usage = None
2055-
if complete_pb.HasField('usage'):
2056-
usage = ConversationUsage(
2057-
prompt_tokens=complete_pb.usage.prompt_tokens,
2058-
completion_tokens=complete_pb.usage.completion_tokens,
2059-
total_tokens=complete_pb.usage.total_tokens
2060-
)
2061-
2062-
complete = ConversationStreamComplete(
2063-
context_id=complete_pb.contextID if complete_pb.HasField('contextID') else None,
2064-
usage=usage
2065-
)
2066-
2052+
complete = ConversationStreamComplete.from_proto(complete_pb)
20672053
yield ConversationStreamResponse(complete=complete)
20682054

20692055
except RpcError as err:
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
apiVersion: dapr.io/v1alpha1
2+
kind: Component
3+
metadata:
4+
name: echo
5+
spec:
6+
type: conversation.echo
7+
version: v1
8+
metadata:
9+
- name: key
10+
value: testkey
11+
- name: timeout
12+
value: 30s

tools/run_dapr_dev.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,13 @@
2929

3030
SCRIPT_DIR = Path(__file__).parent
3131
REPO_ROOT = SCRIPT_DIR.parent
32+
33+
# make sure dapr repo is in the right place
3234
DAPR_REPO = REPO_ROOT.parent / "dapr"
35+
if not DAPR_REPO.exists():
36+
print(f"❌ Error: Dapr repository not found at {DAPR_REPO}")
37+
print("Please clone the dapr repository at ../dapr relative to python-sdk and dapr-agents")
38+
sys.exit(1)
3339

3440

3541
def load_env_file():
@@ -84,6 +90,7 @@ def load_env_file():
8490
def get_provider_key(provider, env_vars, providers_alt_keys):
8591
"""Get the API key for a provider."""
8692
provider_prefixes = providers_alt_keys.get(provider, [provider])
93+
value = None
8794
for prefix in provider_prefixes:
8895
key = f"{prefix}_API_KEY"
8996
if key in env_vars:
@@ -256,12 +263,6 @@ def run_daprd(args):
256263

257264
# Load environment variables from .env file
258265
env_vars = load_env_file()
259-
260-
# Set environment variables for the process
261-
# Some components need the environment variables to be available at runtime
262-
for key, value in env_vars.items():
263-
os.environ[key] = value
264-
265266
# Special mapping for Gemini - it expects GEMINI_API_KEY but we have GOOGLE_AI_API_KEY
266267
if "GOOGLE_AI_API_KEY" in env_vars:
267268
os.environ["GEMINI_API_KEY"] = env_vars["GOOGLE_AI_API_KEY"]
@@ -309,6 +310,7 @@ def run_daprd(args):
309310

310311
try:
311312
# Run the sidecar with explicit environment inheritance
313+
# Set environment variables for the process
312314
env = os.environ.copy()
313315
env.update(env_vars)
314316
subprocess.run(cmd, check=True, env=env)

0 commit comments

Comments
 (0)